Home
last modified time | relevance | path

Searched defs:tensor_indexes_shapes (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dget_tuple_index_info.cc68 static ShapeVector BroadCastShape(const std::vector<ShapeVector> &tensor_indexes_shapes) { in BroadCastShape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.h408 static ShapeVector BroadCastShape(const std::vector<ShapeVector> &tensor_indexes_shapes) { in BroadCastShape()
Dtensor_index_py.cc335 …const std::vector<int64_t> &tensor_positions, const std::vector<ShapeVector> &tensor_indexes_shape… in GenerateIndexInfoFromTupleOfMixedTensors()
719 std::vector<ShapeVector> tensor_indexes_shapes; in TensorGetitemByTuple() local
1027 std::vector<ShapeVector> tensor_indexes_shapes; in GenerateIndicesFromTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/
D_constexpr_utils.py586 def generate_index_info_from_tuple_of_mixed_tensors(tensor_positions, tensor_indexes_shapes, argument