Home
last modified time | relevance | path

Searched defs:MatrixInfo (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dflatten.cc28 struct MatrixInfo { struct
29 std::vector<int64_t> matrix_shape;
30 DataType matrix_type;
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/
Dreshape_kernels.h27 struct MatrixInfo { struct
38 MatrixInfo matrix_info_; argument
Dexpand_dims_kernels.h26 struct MatrixInfo { struct
37 MatrixInfo matrix_info_; argument
Dsqueeze_kernels.h27 struct MatrixInfo { struct
38 MatrixInfo matrix_info_; argument
/third_party/skia/third_party/externals/tint/src/transform/
Ddecompose_strided_matrix.cc35 struct MatrixInfo { struct
37 uint32_t stride = 0;
39 const sem::Matrix* matrix = nullptr;
43 const ast::Array* array(ProgramBuilder* b) const { in array()
49 bool operator==(const MatrixInfo& info) const { in operator ==()
53 struct Hasher {
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dmatmul_struct.h24 typedef struct MatrixInfo { struct
31 } MatrixInfo; argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/
DCommonDataType.h143 struct MatrixInfo { struct
160 MatrixInfo A; argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dmatmul_fp32_base.h81 struct MatrixInfo { struct
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dcommon_utils.h62 explicit MatrixInfo(size_t max_index, const ShapeVector &matrix_shapes) in MatrixInfo() function