Home
last modified time | relevance | path

Searched defs:tensor_shape_ (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dfftshift_cpu_kernel.h53 std::vector<int64_t> tensor_shape_; variable
Dirfft_cpu_kernel.h58 std::vector<int64_t> tensor_shape_; variable
Dirfft_grad_cpu_kernel.h58 std::vector<int64_t> tensor_shape_; variable
Drfft_grad_cpu_kernel.h58 std::vector<int64_t> tensor_shape_; variable
Dfftbase_cpu_kernel.h58 std::vector<int64_t> tensor_shape_; variable
Dfftnbase_cpu_kernel.h69 std::vector<int64_t> tensor_shape_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/
Ddata_schema.h114 std::unique_ptr<TensorShape> tensor_shape_; // The fixed shape (if given by user) variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/core/
Ddevice_tensor_ascend910b.h88 TensorShape tensor_shape_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.h64 Shape tensor_shape_; variable
Dtensor_layout.h180 Arrangement tensor_shape_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dmbuf_receive_manager.h69 ShapeVector tensor_shape_; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/
Deigen_tensor.h174 ShapeVector tensor_shape_; variable
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Darray_func.py118 tensor_shape_ = P.TensorShape() variable
Dnn_func.py93 tensor_shape_ = P.TensorShape() variable