Home
last modified time | relevance | path

Searched defs:subgraph (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc46 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
60 void SubgraphBuilder::BuildAddSubgraph(Subgraph* subgraph) { in BuildAddSubgraph()
90 void SubgraphBuilder::BuildMulSubgraph(Subgraph* subgraph) { in BuildMulSubgraph()
120 void SubgraphBuilder::BuildPadSubgraph(Subgraph* subgraph) { in BuildPadSubgraph()
148 void SubgraphBuilder::BuildIfSubgraph(Subgraph* subgraph) { in BuildIfSubgraph()
186 void SubgraphBuilder::BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs) { in BuildLessEqualCondSubgraph()
217 void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) { in BuildAccumulateLoopBodySubgraph()
261 void SubgraphBuilder::BuildPadLoopBodySubgraph(Subgraph* subgraph, in BuildPadLoopBodySubgraph()
311 void SubgraphBuilder::BuildWhileSubgraph(Subgraph* subgraph) { in BuildWhileSubgraph()
349 void SubgraphBuilder::CreateConstantInt32Tensor(Subgraph* subgraph, in CreateConstantInt32Tensor()
Dtest_util.cc105 auto subgraph = CreateSubGraph(builder_, tensors, inputs, outputs, operators); in BuildInterpreter() local
/external/tensorflow/tensorflow/lite/
Dnnapi_delegate_disabled.cc38 TfLiteStatus NNAPIDelegate::BuildGraph(Subgraph* subgraph) { in BuildGraph()
42 TfLiteStatus NNAPIDelegate::Invoke(Subgraph* subgraph) { return kTfLiteError; } in Invoke()
Dinterpreter.cc113 Subgraph* subgraph = in AddSubgraphs() local
207 for (auto& subgraph : subgraphs_) { in SetNumThreads() local
220 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32() local
229 for (auto& subgraph : subgraphs_) { in SetCancellationFunction() local
235 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegate() local
275 for (auto& subgraph : subgraphs_) subgraph->SetProfiler(profiler); in SetProfiler() local
Dnnapi_delegate.cc115 TfLiteStatus addTensorOperands(tflite::Subgraph* subgraph, in addTensorOperands()
215 tflite::Subgraph* subgraph, ANeuralNetworksModel* nn_model, in AddOpsAndParams()
698 TfLiteStatus NNAPIDelegate::BuildGraph(Subgraph* subgraph) { in BuildGraph()
771 TfLiteStatus NNAPIDelegate::Invoke(Subgraph* subgraph) { in Invoke()
Dmodel.cc249 Subgraph* subgraph) { in ParseNodes()
355 Subgraph* subgraph) { in ParseTensors()
526 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index]; in operator ()() local
Dinterpreter.h461 Subgraph* subgraph(int subgraph_index) { in subgraph() function
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.h33 SubgraphQuantizer(ModelT* model, SubGraphT* subgraph, in SubgraphQuantizer()
Dquantize_weights.cc50 const SubGraphT* subgraph, in GetTensorConsumers()
126 bool CheckAllOpInputsQuantized(const SubGraphT* subgraph, const OperatorT* op, in CheckAllOpInputsQuantized()
151 SubGraphT* subgraph = model->subgraphs.at(0).get(); in InsertQuantizableInputTensorsFromOperator() local
267 SubGraphT* subgraph = model->subgraphs.at(0).get(); in QuantizeWeightsInternal() local
Dsubgraph_quantizer_test.cc71 auto subgraph = model.subgraphs[0].get(); in TEST() local
170 auto subgraph = model.subgraphs[0].get(); in TEST() local
282 auto subgraph = model.subgraphs[0].get(); in TEST() local
339 auto subgraph = model.subgraphs[0].get(); in TEST() local
396 auto subgraph = model.subgraphs[0].get(); in TEST() local
451 auto subgraph = model.subgraphs[0].get(); in TEST() local
Dquantize_model.cc39 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeModel() local
Dquantize_weights_test.cc76 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in IsModelInputOrOutput() local
95 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in GetProducerOpCode() local
Dquantize_model_test.cc125 for (const auto& subgraph : model_.subgraphs) { in TEST_F() local
/external/dagger2/compiler/src/main/java/dagger/internal/codegen/
DSubcomponentWriter.java58 BindingGraph subgraph) { in SubcomponentWriter()
70 private static String subcomponentSimpleName(BindingGraph subgraph) { in subcomponentSimpleName()
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc58 std::unique_ptr<Graph> subgraph( in PartitionFunctionGraph() local
73 Graph* subgraph, std::vector<int>* arg_indices, in UpdateArgAndRetvalMetadata()
Deval_const_tensor.cc331 Graph subgraph(&ops); in EvaluateConstantTensor() local
Dpartitioning_utils_test.cc94 void SubGraph(Graph* subgraph, DataType dtype, int arg_index, int ret_index) { in SubGraph()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_reader.cc39 const auto& subgraph = model->subgraphs[0]; in AddCalibrationToModel() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc1425 Subgraph& subgraph = subgraphs_[func_id]; in CopySubgraphNodes() local
1544 Subgraph& subgraph = entry.second; in SplitIntoSubgraphs() local
1551 Subgraph& subgraph = entry.second; in SplitIntoSubgraphs() local
1572 Subgraph& subgraph = subgraph_entry.second; in BuildFunctionDefs() local
1613 Subgraph& subgraph = subgraph_entry.second; in AddOutsideCompilationHostIONodes() local
1768 const string& subgraph = ancestors.first; in AddCallNodeDependencies() local
1814 Subgraph& subgraph = subgraph_entry.second; in AddEdgesToOutputGraph() local
2087 string subgraph; member
2128 string subgraph; member
2227 Subgraph& subgraph = subgraphs_[subgraph_name]; in FindClusterDependencies() local
[all …]
/external/tensorflow/tensorflow/core/graph/
Dsubgraph_test.cc110 Graph* subgraph = new Graph(OpRegistry::Global()); in Subgraph() local
376 Graph* subgraph = new Graph(OpRegistry::Global()); in BM_SubgraphHelper() local
Dsubgraph.cc37 namespace subgraph { namespace
/external/bcc/tools/
Ddeadlock_detector.py102 def subgraph(self, nodes): member in DiGraph
/external/tensorflow/tensorflow/lite/tools/
Dverifier_test.cc107 auto subgraph = std::vector<Offset<SubGraph>>({CreateSubGraph( in FinishModel() local
253 auto subgraph = std::vector<Offset<SubGraph>>( in TEST() local
Dverifier.cc195 bool VerifySubGraphConsistency(const Model& model, const SubGraph& subgraph, in VerifySubGraphConsistency()
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc.cc208 GraphDef subgraph; in IdentifyLoops() local

12