Home
last modified time | relevance | path

Searched defs:tensor_indices_shapes (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dget_tuple_index_info.cc120 … size_t *slice_nums, std::vector<ShapeVector> *tensor_indices_shapes) { in GetSliceShape()
166 std::vector<ShapeVector> tensor_indices_shapes; in ConstGetTupleIndexInfo() local
274 std::vector<ShapeVector> tensor_indices_shapes; in GetTupleIndexInfoInferInner() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dget_tuple_index_info_cpu_kernel.cc73 std::vector<ShapeVector> tensor_indices_shapes; in LaunchKernel() local