Searched defs:indices_node (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | adjust_resize_dims_pass.cc | 183 auto indices_node = in CreateGatherCNode() local
238 …auto indices_node = BuildIntVecParameterNode(func_graph, {kIndices_1, kIndices_1, kIndices_1, kInd… in CreateMulNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/ |
| D | tensor_index.cc | 753 AnfNodePtr indices_node = NewCNode(remake_tuple_inputs, res_graph_); in RemakeTupleIndex() local
766 …AnfNodePtr indices_node = res_graph_->NewCNode({NewValueNode(concat_prim), remake_tuple, axis_node… in RemakeTupleIndex() local
1125 AnfNodePtr indices_node = NewCNode(remake_tuple_inputs, res_graph_); in SetItemByTuple() local
1139 …AnfNodePtr indices_node = res_graph_->NewCNode({NewValueNode(concat_prim), remake_tuple, axis_node… in SetItemByTuple() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 1241 auto indices_node = BuildIntVecParameterNode(func_graph, indices, cnode_name + "_indices"); in GenGatherNode() local
1269 const AnfNodePtr &indices_node, const std::string &cnode_name, in GenGatherNodeDynamicIndex()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_utils.cc | 2298 auto indices_node = Common::CreateValueNodeByValue(csr_tensor->GetIndices()); in BuildSparseTensorNode() local
2308 auto indices_node = Common::CreateValueNodeByValue(coo_tensor->GetIndices()); in BuildSparseTensorNode() local
|