|
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. |