| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_kernel_util.cc | 54 if (kernel_is_input && !lite::IsContain(input_nodes, kernel)) { in SubgraphInputNodes()
72 if (!lite::IsContain(output_nodes, kernel)) { in SubgraphOutputNodes()
79 !lite::IsContain(output_nodes, kernel)) { in SubgraphOutputNodes()
94 if (!lite::IsContain(input_tensors, in_node_in_tensor)) { in SubgraphInputTensors()
109 if (!lite::IsContain(input_tensors, in_node_in_tensor)) { in SubgraphInputTensors()
127 if (!lite::IsContain(output_tensors, out_kernel_out_tensor)) { in SubgraphOutputTensors()
143 if (!lite::IsContain(output_tensors, out_kernel_out_tensor)) { in SubgraphOutputTensors()
171 if (lite::IsContain(*kernels, const_cast<kernel::LiteKernel *>(next_kernel))) { in TopologicalSortKernels()
176 return lite::IsContain(*kernels, const_cast<kernel::LiteKernel *>(in_kernel)); in TopologicalSortKernels()
|
| D | lite_kernel.cc | 31 if (IsContain(scope_tensors, in_tensor)) { in IsReady()
|
| D | lite_kernel.h | 301 if (!lite::IsContain(this->in_kernels_, kernel)) { in AddInKernel()
307 if (!lite::IsContain(this->out_kernels_, kernel)) { in AddOutKernel()
|
| D | lite_session.cc | 241 if (IsContain(model_input_indices, i)) { in ConvertTensors()
249 if (IsContain(model_output_indices, i)) { in ConvertTensors()
598 if (IsContain(out_tensors, tensor)) { in IsIsolatedSubGraph()
|
| D | scheduler.cc | 1383 if (IsContain(node->input_indices_, output_index)) { in IsControlFlowPattern()
1455 kernel->set_is_model_output(IsContain(graph_output_node_indexes_, size_t(node_index))); in ScheduleSubGraphToKernels()
1658 if ((IsContain(node->output_indices_, true_branch_output_index) || in InferSwitchShape()
1659 IsContain(node->output_indices_, false_branch_output_index)) && in InferSwitchShape()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | subgraph_tensor_pass.cc | 32 if (IsContain<uint32_t>(node->inputIndex, tensor_idx)) { in IsUsing()
35 if (IsContain<uint32_t>(node->outputIndex, tensor_idx)) { in IsUsing()
40 if (IsContain<uint32_t>(subgraph->inputIndices, tensor_idx)) { in IsUsing()
43 if (IsContain<uint32_t>(subgraph->outputIndices, tensor_idx)) { in IsUsing()
|
| D | subgraph_node_pass.cc | 56 return (tensors_indices.count(idx) > 0) || IsContain(subgraph->inputIndices, idx); in IsNodeInputInSubgraph()
64 return (tensors_indices.count(idx) > 0) || IsContain(subgraph->outputIndices, idx); in IsNodeOutputInSubgraph()
99 if (!IsContain(new_nodes, *it)) { in Run()
116 if (!IsContain(old_nodes_, new_nodes[i])) { in Run()
136 if (IsContain(contain_node_output_subgraphs, subgraph)) { in Run()
|
| D | topological_sort_pass.cc | 61 if (IsContain(subgraph_node_indices, (unsigned int)(post_node_idx))) { in Run()
88 … [&](size_t input_idx) { return IsContain(sinked_tensor_idxes, size_t(input_idx)); }); in IsNodeNonDepend()
|
| D | infershape_pass.cc | 418 if (IsContain(node->outputIndex, true_branch_output_index) && in InferSwitchNode()
429 if (IsContain(node->outputIndex, false_branch_output_index) && in InferSwitchNode()
459 if (IsContain(node->outputIndex, call_first_input_index) && in InferCallNode()
469 …if (IsContain(node->outputIndex, call_first_input_index) && node->primitive->value.type == Primiti… in InferCallNode()
567 if (IsContain(subgraph->inputIndices, i) || !graph->allTensors.at(i)->data.empty()) { in InitSearchTensor()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/fusion/ |
| D | fusion_pass.cc | 141 if (IsContain(sinkIdes, nodeIdx)) { in MatchOnePattern()
149 if (IsContain(outputOp->types, node->primitive->value.type)) { in MatchOnePattern()
164 if (IsContain(sinkIdes, nodeIdx)) { in MatchOnePattern()
218 if (!IsContain(matchedNodeIdxes, inputNodePostNodeIdx)) { in CheckMatch()
226 if (!IsContain(matchedNodeIdxes, outputNodePreNodeIdx)) { in CheckMatch()
304 if (IsContain(sinkIdes, nodeIdx) && !IsContain(pathSinkIdes, nodeIdx)) { in CheckMatchParams()
309 if (!target->isPlaceHold && !IsContain(target->types, scope->primitive->value.type)) { in CheckMatchParams()
|
| D | fusion_pattern.cc | 110 if (IsContain(ids, patternOp->id)) { in Finish()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/ |
| D | caffe_inspector.cc | 52 if (!IsContain(layerTops, iter)) { in FindGraphInputsAndOutputs()
57 if (layerBottoms.find(iter) == layerBottoms.end() && !IsContain(graphOutput, iter)) { in FindGraphInputsAndOutputs()
74 if (!IsContain(layerTops, layer.top(j))) { in SetLayerTopsAndBottoms()
|
| /third_party/mindspore/mindspore/lite/src/common/ |
| D | graph_util.cc | 42 if (!IsContain<size_t>(ret, j)) { in GetGraphInputNodes()
61 if (!IsContain<size_t>(ret, j)) { in GetGraphOutputNodes()
94 return IsContain(packed_ops, op_type); in IsPackedOp()
|
| D | utils.h | 51 bool IsContain(const std::vector<T> &vec, T element) { in IsContain() function
90 if (!IsContain(*vec, dstElement)) { in VectorReplace()
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_utils.cc | 143 if (!IsContain(valid_type, tensor->data_type())) { in CastTensor()
148 if (!IsContain(valid_type, dst_data_type)) { in CastTensor()
160 if (IsContain(fp32_type, tensor->data_type()) && dst_data_type == kNumberTypeFloat16) { in CastTensor()
196 if (!IsContain(valid_type, tensor->data_type())) { in ScaleTensor()
|
| D | train_export.cc | 44 if (IsContain<uint32_t>(input_indexes, tensor_idx)) { in GetLinkedPostIdx()
477 if (IsContain(subgraph_node_indices, (unsigned int)(post_node_idx))) { in TopologicalSort()
504 … [&](size_t input_idx) { return IsContain(sinked_tensor_idxes, size_t(input_idx)); }); in IsNodeNonDepend()
|
| D | train_session.cc | 939 if (IsContain(inference_kernels[i]->out_tensors(), kernel_in_tensors[j])) { in FindUseInTensorKernel()
957 return IsContain(export_output_tensor_names, out_tensor->tensor_name()); in FindExportKernels()
976 if (!IsContain(*export_kernels, kernel)) { in FindExportKernels()
|
| /third_party/mindspore/mindspore/lite/src/runtime/ |
| D | runtime_pass.cc | 89 if (IsContain(Nc4hw4FormatOutOpList, start_kernel->type()) == false) { in Nc4hw4PassMatch()
109 if (IsContain(Nc4hw4FormatInOpList, end_kernel->type()) == false) { in Nc4hw4PassMatch()
|
| /third_party/mindspore/mindspore/lite/src/ops/populate/ |
| D | populate_register.h | 57 if (lite::IsContain(string_op, static_cast<schema::PrimitiveType>(type))) { in GetParameterCreator()
|
| /third_party/mindspore/mindspore/lite/tools/common/ |
| D | graph_util.cc | 142 if (IsContain<uint32_t>(outputIndexes, tensorIdx)) { in GetLinkedPreIdx()
157 if (IsContain<uint32_t>(inputIndexes, tensorIdx)) { in GetLinkedPostIdx()
417 if (!IsContain(node->inputIndex, inTensorIdx)) { in ReplaceTensorOfNode()
589 bool is_output_index = IsContain(graphT->outputIndex, postTensorIdx); in InsertNodeAfter()
683 if (IsContain(subgraph_indices, input_idx)) { in SetSubgraphTensorIndices()
690 if (IsContain(subgraph_indices, output_idx)) { in SetSubgraphTensorIndices()
|
| D | tensor_util.cc | 250 if (IsContain(node->inputIndex, tensorIdx)) { in GetRefCount()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_inputs_adjust.cc | 192 if (lite::IsContain(single_input_ops, primitive->name()) || CheckResize(cnode)) { in Run()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 68 if (!IsContain(conv_types_, primitive_c->name())) { in CanConvOpQuantized()
156 if (!IsContain(mul_types_, primitive_c->name())) { in CanMulOpQuantized()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_lstm_cell_fusion.cc | 348 if (!lite::IsContain(valid_indexes, index)) { in CheckReferencedOutputs()
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | anf_exporter.cc | 770 if (IsContain(graph_inputs_, input_node->cast<AnfNodePtr>()) && in SetOpInputNode()
|