Home
last modified time | relevance | path

Searched defs:tensor_dims (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dremove_expanded_dims.cc46 std::vector<bool> tensor_dims(tensor_index_ndim, true); in RemNotExpandedDims() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/
Dmodel.cpp567 std::vector<int64_t> tensor_dims(dim_size); in Java_com_mindspore_Model_resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc847 std::vector<bool> tensor_dims(tensor_index_ndim, true); in RemNotExpandedDims() local