Searched defs:tile_node (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/function/ |
| D | func_pass.cc | 175 auto tile_node = func_builder_->EmitOp(tile_primitive, tile_inputs); in CreateTile() local
184 NodePtr CreateRealDiv(const NodePtr &tile_node) { in CreateRealDiv()
216 auto tile_node = CreateTile(inputs, dout); in CreateMulInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 370 auto tile_node = pass.NewCNode(tile_inputs, graph); in CreateTile() local
386 …ealDiv(const FuncGraphPtr &graph, const CNodePtr &sparse_softmax_node, const AnfNodePtr &tile_node, in CreateRealDiv()
525 auto tile_node = CreateTile(graph, sparse_softmax_node_grad, mul_node, pass); in CreateMulInput() local
618 auto tile_node = CreateTile(graph, sparse_softmax_node_grad, mul_node, *this); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ir/ |
| D | ir_pass.cc | 331 auto tile_node = ir_bprop_->ad_param()->tape_->FuncGraph::NewCNode(tile_inputs); in CreateTile() local
345 CNodePtr CreateRealDiv(const CNodePtr &sparse_softmax_node, const AnfNodePtr &tile_node) { in CreateRealDiv()
401 auto tile_node = CreateTile(sparse_softmax_cnode, mul_node); in CreateMulInput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | flash_sp.cc | 310 auto tile_node = tensor_node->func_graph()->NewCNode(tile_inputs); in NewTileNode() local
|