Home
last modified time | relevance | path

Searched defs:TensorMap (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/
Ddataset_iterator.h33 using TensorMap = std::unordered_map<std::string, std::shared_ptr<Tensor>>; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.h34 using TensorMap = Shape; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dnode_strategy.proto39 message TensorMap { message
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/transform/graph_ir/
Dtypes.h53 using TensorMap = mindspore::HashMap<std::string, std::shared_ptr<MeTensor>>; variable
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/
Dutil_layout_gen_test.cc153 std::vector<std::tuple<DeviceArrangement, TensorMap, TensorShape>> *layout_list) { in GenerateValidLayoutByDeviceSizeAndTensorSize() argument