Home
last modified time | relevance | path

Searched refs:node_index (Results 1 – 25 of 130) sorted by relevance

123456

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dset_fracz_group_attr.cc46 for (auto node_index : manager->node_users()[node]) { in GetOutputItem() local
47 … if (AnfAlgo::CheckPrimitiveType(node_index.first, prim::kPrimDepend) && node_index.second == 1) { in GetOutputItem()
48 (void)depend_nodes.emplace_back(node_index.first); in GetOutputItem()
49 } else if (AnfAlgo::CheckPrimitiveType(node_index.first, prim::kPrimTupleGetItem)) { in GetOutputItem()
50 auto getitem_cnode = node_index.first->cast<CNodePtr>(); in GetOutputItem()
140 [](KernelWithIndex node_index) { in GetCNodeNeighborFraczNodes() argument
141 return KernelWithIndex{node_index.first, node_index.second - 1}; in GetCNodeNeighborFraczNodes()
156 [](KernelWithIndex node_index) { in GetNeighborFraczNodes() argument
157 return KernelWithIndex{node_index.first, node_index.second - 1}; in GetNeighborFraczNodes()
172 [](KernelWithIndex node_index) { in GetNeighborFraczNodes() argument
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_parse_graph.cc185 void Eliminate_Aux(const size_t node_index, const std::shared_ptr<Graph> &graph, in Eliminate_Aux() argument
188 eli.push_back(node_index); in Eliminate_Aux()
189 for (size_t i = 0; i < (size_t)graph->nodes[node_index].node_out.size(); i++) { in Eliminate_Aux()
190 eli.push_back(graph->nodes[node_index].node_out[i]); in Eliminate_Aux()
194 for (size_t i = 0; i < graph->nodes[node_index].node_in.size(); i++) { in Eliminate_Aux()
195 auto *incoming_outputs = &graph->nodes[graph->nodes[node_index].node_in[i]].node_out; in Eliminate_Aux()
196 auto it = find(incoming_outputs->begin(), incoming_outputs->end(), node_index); in Eliminate_Aux()
199 …incoming_outputs->insert(it, graph->nodes[node_index].node_out.begin(), graph->nodes[node_index].n… in Eliminate_Aux()
203 for (size_t i = 0; i < graph->nodes[node_index].node_in_aux.size(); i++) { in Eliminate_Aux()
204 auto *aux_incoming_outputs = &graph->nodes[graph->nodes[node_index].node_in_aux[i]].node_out; in Eliminate_Aux()
[all …]
/third_party/glib/glib/
Dghash.c465 guint node_index; in g_hash_table_lookup_node() local
478 node_index = g_hash_table_hash_to_index (hash_table, hash_value); in g_hash_table_lookup_node()
479 node_hash = hash_table->hashes[node_index]; in g_hash_table_lookup_node()
489 …gpointer node_key = g_hash_table_fetch_key_or_value (hash_table->keys, node_index, hash_table->hav… in g_hash_table_lookup_node()
494 return node_index; in g_hash_table_lookup_node()
498 return node_index; in g_hash_table_lookup_node()
503 first_tombstone = node_index; in g_hash_table_lookup_node()
508 node_index += step; in g_hash_table_lookup_node()
509 node_index &= hash_table->mask; in g_hash_table_lookup_node()
510 node_hash = hash_table->hashes[node_index]; in g_hash_table_lookup_node()
[all …]
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Dconv2d_delegate_fp32_coder.cc33 …CPUConvolutionFP32CoderSelect(input_tensors_, output_tensors_, node_, node_index(), target_, schem… in Prepare()
66 … const Model::Node *node, size_t node_index, Target target, in CPUConvolutionFP32CoderSelect() argument
92 …coder = CPUOpCoderCreator<ConvolutionFP32Coder>(in_tensors, out_tensors, node, node_index, target,… in CPUConvolutionFP32CoderSelect()
95 …ke_unique<ConvolutionWinogradFP32Coder>(in_tensors, out_tensors, node, node_index, target, out_uni… in CPUConvolutionFP32CoderSelect()
98 …coder = CPUOpCoderCreator<ConvolutionFP32Coder>(in_tensors, out_tensors, node, node_index, target,… in CPUConvolutionFP32CoderSelect()
105 … size_t node_index, Target target, int schema_version) { in CreateDelegateConv() argument
106 …return CPUOpCoderCreator<ConvDelegateCoder>(in_tensors, out_tensors, node, node_index, target, sch… in CreateDelegateConv()
111 … const Model::Node *node, size_t node_index, Target target, in CPUConvDwFp32CoderCreator() argument
113 …return CPUOpCoderCreator<ConvolutionDepthwiseFP32Coder>(in_tensors, out_tensors, node, node_index,… in CPUConvDwFp32CoderCreator()
119 … const Model::Node *node, size_t node_index, in CPUConv2DFusionFP32CoderCreator() argument
[all …]
Dconv2d_delegate_fp32_coder.h27 const Model::Node *node, size_t node_index, Target target) in ConvDelegateCoder() argument
28 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ConvDelegateCoder()
41 … const Model::Node *node, size_t node_index, Target target,
46 … size_t node_index, Target target, int schema_version);
50 … const Model::Node *node, size_t node_index, Target target,
55 … const Model::Node *node, size_t node_index,
Dassign_add_fp32_coder.h28 const Model::Node *node, size_t node_index, Target target) in AssignAddFP32Coder() argument
29 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in AssignAddFP32Coder()
Dactivation_fp32_coder.h28 const Model::Node *node, size_t node_index, Target target) in ActivationFP32Coder() argument
29 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ActivationFP32Coder()
Dpooling_fp32_coder.h28 const Model::Node *node, size_t node_index, Target target) in PoolingFP32Coder() argument
29 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in PoolingFP32Coder()
Daddn_fp32_coder.h26 const Model::Node *node, size_t node_index, Target target) in AddNFP32Coder() argument
27 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in AddNFP32Coder()
Dsplice_fp32_coder.h25 const Model::Node *node, size_t node_index, Target target) in SpliceFP32Coder() argument
26 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in SpliceFP32Coder()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Drelux_int8_coder.h32 const Model::Node *node, size_t node_index, Target target) in ReluxInt8Coder() argument
33 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ReluxInt8Coder()
51 const Model::Node *node, size_t node_index, Target target) in ReluInt8Coder() argument
52 : ReluxInt8Coder(in_tensors, out_tensors, node, node_index, target) {} in ReluInt8Coder()
67 const Model::Node *node, size_t node_index, Target target) in Relu6Int8Coder() argument
68 : ReluxInt8Coder(in_tensors, out_tensors, node, node_index, target) {} in Relu6Int8Coder()
Dactivation_int8_coder.cc29 … const Model::Node *node, size_t node_index, Target target, in CPUActivationINT8CoderCreator() argument
53 …coder = CPUOpCoderCreator<SigmodInt8Coder>(in_tensors, out_tensors, node, node_index, target, sche… in CPUActivationINT8CoderCreator()
56 …coder = CPUOpCoderCreator<ReluInt8Coder>(in_tensors, out_tensors, node, node_index, target, schema… in CPUActivationINT8CoderCreator()
59 …coder = CPUOpCoderCreator<Relu6Int8Coder>(in_tensors, out_tensors, node, node_index, target, schem… in CPUActivationINT8CoderCreator()
Dconv2d_int8_coder.cc250 … const Model::Node *node, size_t node_index, Target target, in CPUConv2DINT8CoderCreator() argument
272 …coder = CPUOpCoderCreator<Conv2D3x3Int8Coder>(in_tensors, out_tensors, node, node_index, target, s… in CPUConv2DINT8CoderCreator()
274 …coder = CPUOpCoderCreator<Conv2D1x1Int8Coder>(in_tensors, out_tensors, node, node_index, target, s… in CPUConv2DINT8CoderCreator()
276 …coder = CPUOpCoderCreator<Conv2DINT8Coder>(in_tensors, out_tensors, node, node_index, target, sche… in CPUConv2DINT8CoderCreator()
287 … const Model::Node *node, size_t node_index, in CPUConv2DFusionINT8CoderCreator() argument
302 …coder = CPUConv2DINT8CoderCreator(in_tensors, out_tensors, node, node_index, target, schema_versio… in CPUConv2DFusionINT8CoderCreator()
304 …CPUOpCoderCreator<ConvolutionDepthwiseINT8Coder>(in_tensors, out_tensors, node, node_index, target, in CPUConv2DFusionINT8CoderCreator()
Ddetection_post_process_int8_coder.h29 const Model::Node *node, size_t node_index, Target target) in DetectionPostProcessInt8Coder() argument
30 : DetectionPostProcessBaseCoder(in_tensors, out_tensors, node, node_index, target) {} in DetectionPostProcessInt8Coder()
Dreshape_int8_coder.h27 const Model::Node *node, size_t node_index, Target target) in ReshapeInt8Coder() argument
28 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ReshapeInt8Coder()
Dpooling_int8_coder.h30 const Model::Node *node, size_t node_index, Target target) in PoolingInt8Coder() argument
31 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in PoolingInt8Coder()
/third_party/mesa3d/src/gallium/drivers/r300/compiler/
Dradeon_pair_regalloc.c518 unsigned int i, input_node, node_count, node_index; in do_advanced_regalloc() local
532 for (var_ptr = variables, node_index = 0; var_ptr; in do_advanced_regalloc()
533 var_ptr = var_ptr->Next, node_index++) { in do_advanced_regalloc()
539 node_classes[node_index] = ra_state->classes[class_index]; in do_advanced_regalloc()
572 for (node_index = 0; node_index < node_count; node_index++) { in do_advanced_regalloc()
573 ra_set_node_class(graph, node_index, node_classes[node_index]); in do_advanced_regalloc()
577 for (var_ptr = variables, node_index = 0; var_ptr; in do_advanced_regalloc()
578 var_ptr = var_ptr->Next,node_index++) { in do_advanced_regalloc()
582 for (a = var_ptr, b = var_ptr->Next, b_index = node_index + 1; in do_advanced_regalloc()
590 node_index, b_index); in do_advanced_regalloc()
[all …]
/third_party/mindspore/mindspore/lite/src/
Dsub_graph_split.cc252 uint32_t node_index = subgraph.nodes_.front(); in ConvertSubGraphToModel() local
253 LiteGraph::Node *cur_node = model_->graph_.all_nodes_[node_index]; in ConvertSubGraphToModel()
254 new_sub_graph->node_indices_.push_back(node_index); in ConvertSubGraphToModel()
256 …auto iter = find(main_graphs->node_indices_.begin(), main_graphs->node_indices_.end(), node_index); in ConvertSubGraphToModel()
260 VectorErase(&main_graphs->node_indices_, node_index); in ConvertSubGraphToModel()
261 VectorErase(&subgraph.nodes_, node_index); in ConvertSubGraphToModel()
308 bool SearchSubGraph::IsNodeSubGraphHead(uint32_t node_index, const std::vector<uint32_t> &ready_nod… in IsNodeSubGraphHead() argument
309 std::vector<uint32_t> output_indexes = model_->graph_.all_nodes_.at(node_index)->output_indices_; in IsNodeSubGraphHead()
326 bool SearchSubGraph::IsNodeSubGraphHeadWithRoot(uint32_t node_index, const std::vector<uint32_t> &r… in IsNodeSubGraphHeadWithRoot() argument
328 std::vector<uint32_t> output_indexes = node_list_.at(node_index)->output_indices_; in IsNodeSubGraphHeadWithRoot()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc244 auto node_index = GetNodeIndex(); in GetDeviceShape() local
246 …device_shape = trans::TransShapeToDevice(*host_shape, format_, node_index.first, node_index.second… in GetDeviceShape()
254 …device_shape = trans::TransShapeToDevice(*host_shape, format_, node_index.first, node_index.second… in GetDeviceShape()
331 auto node_index = GetNodeIndex(); in SyncDeviceToHostAndConvertFormat() local
336 … trans::TransFormatFromDeviceToHost(format_args, host.data(), node_index.first, node_index.second); in SyncDeviceToHostAndConvertFormat()
351 …k = trans::TransFormatFromDeviceToHost(format_args, host_ptr, node_index.first, node_index.second); in SyncDeviceToHostAndConvertFormat()
440 auto node_index = GetNodeIndex(); in ConvertFormatAndSyncHostToDevice() local
443 …device_shape = trans::TransShapeToDevice(host_shape, format_, node_index.first, node_index.second); in ConvertFormatAndSyncHostToDevice()
446 …device_shape = trans::TransShapeToDevice(host_shape, format_, node_index.first, node_index.second); in ConvertFormatAndSyncHostToDevice()
460 sync_ok = trans::TransFormat(format_args, dst_tmp.data(), node_index.first, node_index.second); in ConvertFormatAndSyncHostToDevice()
[all …]
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/
Dop_coder.h39 const Model::Node *node, size_t node_index, Target target) in OperatorCoder() argument
44 node_index_(node_index) { in OperatorCoder()
69 size_t node_index() const;
121 … size_t node_index, Target target, int schema_version) { in CPUOpCoderCreator() argument
126 std::unique_ptr<T> coder = std::make_unique<T>(in_tensors, out_tensors, node, node_index, target); in CPUOpCoderCreator()
/third_party/boost/boost/lockfree/detail/
Dfreelist.hpp450 index_t node_index = allocate<ThreadSafe>(); in construct() local
451 if (node_index == null_handle()) in construct()
454 T * node = NodeStorage::nodes() + node_index; in construct()
462 index_t node_index = allocate<ThreadSafe>(); in construct() local
463 if (node_index == null_handle()) in construct()
466 T * node = NodeStorage::nodes() + node_index; in construct()
474 index_t node_index = allocate<ThreadSafe>(); in construct() local
475 if (node_index == null_handle()) in construct()
478 T * node = NodeStorage::nodes() + node_index; in construct()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/
Dreshape_base_coder.h27 const Model::Node *node, size_t node_index, Target target) in ReshapeBaseCoder() argument
28 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ReshapeBaseCoder()
Dresize_base_coder.h29 const Model::Node *node, size_t node_index, Target target) in ResizeBaseCoder() argument
30 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ResizeBaseCoder()
Dsoftmax_base_coder.h31 const Model::Node *node, size_t node_index, Target target) in SoftmaxBaseCoder() argument
32 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in SoftmaxBaseCoder()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/
Dreshape_int8_coder.h27 const Model::Node *node, size_t node_index, Target target) in ReshapeInt8Coder() argument
28 : OperatorCoder(in_tensors, out_tensors, node, node_index, target) {} in ReshapeInt8Coder()

123456