blace.ai
|
#include <public_ops.h>
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) | |
Operator to construct an empty tensor.
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.
sizes | The sizes of the tensor. |
order | Tensor order. |
device | Device to construct the tensor on. |
data_type | Tensor daty type. |
color_format | Tensor color format. |