Home
last modified time | relevance | path

Searched defs:end_node (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dstrided_slice_fusion.cc124 auto end_node = nodes.back(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dgraph_builder.cc76 auto end_node = Tensor(end); in StridedSlice() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dsub_graph_split.cc95 auto end_node = model_->graph_.all_nodes_.at(ends.front()); in CheckIsParallelSubGraph() local
302 LiteGraph::Node *end_node = model_->graph_.all_nodes_[end_index]; in ConvertSubGraphToModel() local
1030 for (uint32_t end_node : sub->ends_) { in CheckSubHeadEnd() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dshape_unify_mindir.cc144 auto end_node = CreateScalarValueTuple(func_graph, begin_index + kSizeOne); in CreateStridedSlice() local
Dconvert_pad_v3_paddings.cc71 …auto end_node = CreateValueNodeWithKernelInfo(func_graph, MakeValue(std::vector<int64_t>{index + 1… in CreateStridedSliceNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dpadv3_ge_pass.cc61 auto end_node = GenerateDataValueTuple(func_graph, index + kSizeOne); in CreateStridedSlice() local
Dscalar_op_pass.cc239 auto end_node = GenerateScalarValueTuple(func_graph, begin_index + kSizeOne); in GenerateStridedSlice() local
/third_party/mesa3d/src/compiler/nir/
Dnir_lower_task_shader.c323 nir_cf_node *end_node = &current_block->cf_node; in lower_task_launch_mesh_workgroups() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_recompute.cc152 OrderedSet<AnfNodePtr> GetLongTermNodes(const AnfNodePtrList &nodes, const AnfNodePtr &end_node, in GetLongTermNodes()
227 AnfNodePtrList AutoRecompute::Filter(const AnfNodePtr &source_node, const AnfNodePtr &end_node, int… in Filter()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dfold_pipeline_split_utils.cc122 auto end_node = GetPreNode(temp_node); in InsertVirtualFoldPipelineEndNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/
Ddvm_kernel_build.cc548 AnfNodePtr end_node = graph->get_return(); in Construct() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dfallback.cc871 auto end_node = in GeneratePyInterpretNodeFromMetaFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dfallback_rewriter.cc2161 AnfNodePtr end_node; in ConvertValueSlice() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc3606 auto end_node = dyn_cast<ValueNode>(node->input(kFourNum)); in ExportPrimTensorCopySlices() local