Home
last modified time | relevance | path

Searched refs:index_node (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dbatchnorm_to_bninfer.cc47 bool CheckIndex(const AnfNodePtr &index_node) { in CheckIndex() argument
48 MS_EXCEPTION_IF_NULL(index_node); in CheckIndex()
49 if (!IsValueNode<Int64Imm>(index_node)) { in CheckIndex()
52 ValueNodePtr value_node = index_node->cast<ValueNodePtr>(); in CheckIndex()
91 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in NeedFusion() local
92 MS_EXCEPTION_IF_NULL(index_node); in NeedFusion()
93 if (!CheckIndex(index_node)) { in NeedFusion()
Dbatchnormgrad_to_bninfergrad.cc49 bool CheckIndex(const AnfNodePtr &index_node) { in CheckIndex() argument
50 MS_EXCEPTION_IF_NULL(index_node); in CheckIndex()
51 if (!IsValueNode<Int64Imm>(index_node)) { in CheckIndex()
54 ValueNodePtr value_node = index_node->cast<ValueNodePtr>(); in CheckIndex()
93 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in NeedFusion() local
94 MS_EXCEPTION_IF_NULL(index_node); in NeedFusion()
95 if (!CheckIndex(index_node)) { in NeedFusion()
Dfused_batch_norm_fusion.cc226 AnfNodePtr index_node = tuple_getitem_cnode->input(kInputNodeOutputIndexInTupleGetItem); in Process() local
227 MS_EXCEPTION_IF_NULL(index_node); in Process()
228 auto value_node = index_node->cast<ValueNodePtr>(); in Process()
/third_party/f2fs-tools/fsck/
Dsegment.c160 void *index_node = NULL; in f2fs_read() local
189 if (index_node) in f2fs_read()
190 free(index_node); in f2fs_read()
191 index_node = (dn.node_blk == dn.inode_blk) ? in f2fs_read()
224 if (index_node) in f2fs_read()
225 free(index_node); in f2fs_read()
243 void* index_node = NULL; in f2fs_write() local
268 if (index_node) in f2fs_write()
269 free(index_node); in f2fs_write()
270 index_node = (dn.node_blk == dn.inode_blk) ? in f2fs_write()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dgetitem_tuple.cc51 AnfNodePtr index_node = tuple_getitem->input(kInputNodeOutputIndexInTupleGetItem); in Process() local
52 MS_EXCEPTION_IF_NULL(index_node); in Process()
53 if (IsValueNode<Int64Imm>(index_node)) { in Process()
54 auto value_node = index_node->cast<ValueNodePtr>(); in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc117 const std::shared_ptr<ops::Conv2DFusion> &ori_conv_prim, int64_t index_node, in CalSplitInShape() argument
122 MS_ASSERT(node_in_out_shapes.size() > index_node); in CalSplitInShape()
123 auto in_out_shape = node_in_out_shapes.at(index_node); in CalSplitInShape()
142 auto split_axis_dim = (*split_axis_inputs_shape)[index_node][index] - 1; in CalSplitInShape()
161 auto split_axis_reduce_dim = (*split_axis_reduce_inputs_shape)[index_node][index] - 1; in CalSplitInShape()
225 size_t index_node = 0; in UpdateSplitInfo() local
227 while (index_node < node_size) { in UpdateSplitInfo()
229 auto out_node_name = conv_nodes[index_node]->fullname_with_scope(); in UpdateSplitInfo()
236 index_node++; in UpdateSplitInfo()
257 index_node = 0; in UpdateSplitInfo()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbatch_norm_bert_fission.cc48 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in GetBatchNormOutputs() local
49 MS_EXCEPTION_IF_NULL(index_node); in GetBatchNormOutputs()
50 auto value_node = index_node->cast<ValueNodePtr>(); in GetBatchNormOutputs()
Dbatch_norm_grad_infer_fission.cc43 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in CheckOutputsIndex() local
44 MS_EXCEPTION_IF_NULL(index_node); in CheckOutputsIndex()
45 auto value_node = index_node->cast<ValueNodePtr>(); in CheckOutputsIndex()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dupdate_state_formatter.cc189 auto index_node = output_node->input(index); in FindAllOutputs() local
193 auto DependsOnIndexNode = [&index_node](const AnfNodePtr &out_node) -> bool { in FindAllOutputs()
195 auto IncludeFunc = [&result, &index_node](const AnfNodePtr &node) { in FindAllOutputs()
196 if (node == index_node) { in FindAllOutputs()
Doptimize_assign.cc114 auto index_node = getitem->cast<CNodePtr>()->input(kInputNodeOutputIndexInTupleGetItem); in GetitemIndex() local
115 auto value_ptr = GetValueNode(index_node); in GetitemIndex()
Ddecrease_transfer_precision.cc38 auto index_node = getitem->cast<CNodePtr>()->input(kInputNodeOutputIndexInTupleGetItem); in ObtainGetItemIndex() local
39 auto value_ptr = GetValueNode(index_node); in ObtainGetItemIndex()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_grad_fusion.cc54 auto index_node = tuple_getiterm_cnode->input(kInputNodeOutputIndexInTupleGetItem); in GetBatchNormOutputs() local
55 MS_EXCEPTION_IF_NULL(index_node); in GetBatchNormOutputs()
56 auto value_node = index_node->cast<ValueNodePtr>(); in GetBatchNormOutputs()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dredundant_op_remove_pass.cc349 auto index_node = cnode->inputs()[kInputIndexTwo]->cast<ParameterPtr>(); in RemoveInvalidTransposeOp() local
350 if (index_node == nullptr) { in RemoveInvalidTransposeOp()
353 auto tensor_info = std::dynamic_pointer_cast<tensor::Tensor>(index_node->default_param()); in RemoveInvalidTransposeOp()
/third_party/python/Lib/lib2to3/
Dfixer_util.py77 def Subscript(index_node): argument
80 index_node,
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dauto_monad.cc527 auto &index_node = cnode->inputs().at(index_input); in TraceTupleGetItemEffectInfo() local
528 auto index_value = GetValueNode<Int64ImmPtr>(index_node); in TraceTupleGetItemEffectInfo()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.cc1551 auto index_node = node->inputs()[TUPLE_GET_ITEM_INDEX]; in TraceTupleGetItem() local
1552 if (!index_node->isa<ValueNode>()) { in TraceTupleGetItem()
1556 *index = LongToUlong(GetValue<int64_t>(GetValueNode(index_node))); in TraceTupleGetItem()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dfull_quant_quantizer.cc801 auto index_node = cnode->input(THIRD_INPUT); in QuantNode() local
802 auto index_value_node = index_node->cast<mindspore::ValueNodePtr>(); in QuantNode()