Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_index_py.cc507 std::vector<TensorPtr> *tensor_indexes, in TensorGetitemByTupleParseTensorIndex()
665 std::vector<TensorPtr> tensor_indexes; in TensorGetitemByTuple() local
1023 std::vector<TensorPtr> tensor_indexes; in GenerateIndicesFromTuple() local
1613 …Index::GetItemByTuple(const ShapeVector &data_shape, const std::vector<TensorIndex> &tensor_indexe… in GetItemByTuple()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dmem_swap_scheduler.cc92 std::map<size_t, size_t> tensor_indexes; in GetActionTensors() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dgru_fusion_pass.cc89 auto tensor_indexes = node->inputIndex; in CheckArithmetic() local