blace.ai
blace::ops::OnesOp Class Reference

#include <public_ops.h>

Inheritance diagram for blace::ops::OnesOp:

Public Member Functions

 OnesOp (std::vector< int64_t > sizes, ml_core::OrderEnum order, ml_core::DeviceEnum device, ml_core::DataTypeEnum data_type, ml_core::ColorFormatEnum color_format)
 

Detailed Description

Operator to construct an empty tensor.

Constructor & Destructor Documentation

◆ OnesOp()

blace::ops::OnesOp::OnesOp ( std::vector< int64_t >  sizes,
ml_core::OrderEnum  order,
ml_core::DeviceEnum  device,
ml_core::DataTypeEnum  data_type,
ml_core::ColorFormatEnum  color_format 
)

Operator to construct an empty tensor.

Parameters
sizesThe sizes of the tensor.
orderTensor order.
deviceDevice to construct the tensor on.
data_typeTensor daty type.
color_formatTensor color format.