Home
last modified time | relevance | path

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

123

/external/XNNPACK/src/
Dsubgraph.c24 struct xnn_subgraph* subgraph = NULL; in xnn_create_subgraph() local
62 struct xnn_value* xnn_subgraph_new_internal_value(xnn_subgraph_t subgraph) in xnn_subgraph_new_internal_value()
87 struct xnn_node* xnn_subgraph_new_node(xnn_subgraph_t subgraph) in xnn_subgraph_new_node()
114 xnn_subgraph_t subgraph, in xnn_define_convolution_2d()
267 xnn_subgraph_t subgraph, in xnn_define_depthwise_convolution_2d()
412 xnn_subgraph_t subgraph, in xnn_define_add2()
485 xnn_subgraph_t subgraph, in xnn_define_multiply2()
558 xnn_subgraph_t subgraph, in xnn_define_prelu()
607 xnn_subgraph_t subgraph, in xnn_define_clamp()
651 xnn_subgraph_t subgraph, in xnn_define_hardswish()
[all …]
Dtensor.c19 xnn_subgraph_t subgraph, in xnn_define_tensor_value()
75 xnn_subgraph_t subgraph, in xnn_tensor_get_size()
Druntime.c22 xnn_subgraph_t subgraph, in xnn_create_runtime()
29 xnn_subgraph_t subgraph, in xnn_create_runtime_v2()
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc31 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
45 void SubgraphBuilder::BuildAddSubgraph(Subgraph* subgraph) { in BuildAddSubgraph()
75 void SubgraphBuilder::BuildMulSubgraph(Subgraph* subgraph) { in BuildMulSubgraph()
105 void SubgraphBuilder::BuildPadSubgraph(Subgraph* subgraph) { in BuildPadSubgraph()
133 void SubgraphBuilder::BuildIfSubgraph(Subgraph* subgraph) { in BuildIfSubgraph()
167 void SubgraphBuilder::BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs) { in BuildLessEqualCondSubgraph()
198 void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) { in BuildAccumulateLoopBodySubgraph()
242 void SubgraphBuilder::BuildPadLoopBodySubgraph(Subgraph* subgraph, in BuildPadLoopBodySubgraph()
292 void SubgraphBuilder::BuildWhileSubgraph(Subgraph* subgraph) { in BuildWhileSubgraph()
326 void SubgraphBuilder::CreateConstantInt32Tensor(Subgraph* subgraph, in CreateConstantInt32Tensor()
Dassign_variable.cc56 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval() local
Dread_variable.cc52 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in Eval() local
Dtest_util.cc157 auto subgraph = CreateSubGraph(builder_, tensors, inputs, outputs, operators); in BuildInterpreter() local
273 int CountPartitionsDelegatedTo(Subgraph* subgraph, in CountPartitionsDelegatedTo()
290 Subgraph* subgraph = interpreter->subgraph(i); in CountPartitionsDelegatedTo() local
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc333 TensorType GetTensorType(int32_t idx, const SubGraph* subgraph) { in GetTensorType()
354 const SubGraph* subgraph) { in GetOpSignature()
449 const SubGraph* subgraph = subgraphs->Get(i); in UpdateOpVersion() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc125 int32_t SetInputType(ModelT* model, SubGraphT* subgraph, in SetInputType()
179 int32_t SetOutputType(ModelT* model, SubGraphT* subgraph, in SetOutputType()
244 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetInputAndOutputTypes() local
293 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in ApplyConstraints() local
390 bool IsSubgraphInput(SubGraphT* subgraph, int32_t index) { in IsSubgraphInput()
407 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeOpInput() local
591 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeOpOutput() local
674 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeIntemediateTensors() local
741 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeSharedRange() local
799 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeWeightsInputOutput() local
[all …]
Dquantize_model_test.cc189 for (const auto& subgraph : model_.subgraphs) { in TEST_F() local
203 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_F() local
257 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_F() local
362 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_F() local
441 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_F() local
502 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
603 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
702 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
762 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
824 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
[all …]
Dquantize_weights.cc52 const SubGraphT* subgraph, in GetTensorConsumers()
159 bool CheckAllOpInputsQuantized(const SubGraphT* subgraph, const OperatorT* op, in CheckAllOpInputsQuantized()
187 SubGraphT* subgraph = model->subgraphs.at(0).get(); in InsertQuantizableInputTensorsFromOperator() local
309 const ModelT* model, const SubGraphT* subgraph, in PassQuantizationAndGetConsumers()
356 SubGraphT* subgraph = model->subgraphs.at(0).get(); in QuantizeWeightsInt8() local
479 SubGraphT* subgraph = model->subgraphs.at(0).get(); in QuantizeWeightsFloat16() local
Dquantization_wrapper_utils.cc44 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in IntermediateTensorExists() local
78 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddIntemediateTensorsToFusedOp() local
Dquantization_utils_test.cc472 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
521 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
555 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
587 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
609 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
642 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
684 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
721 auto subgraph = absl::make_unique<SubGraphT>(); in TEST_F() local
742 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
Dmodel_utils.cc107 bool HasBuffer(const ModelT* model, const SubGraphT* subgraph, in HasBuffer()
125 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetOperatorCodeVersion() local
Dquantization_wrapper_utils_test.cc32 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST() local
Dmodel_utils_test.cc46 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST() local
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc178 Subgraph* subgraph = new Subgraph(error_reporter_, external_contexts_, in AddSubgraphs() local
270 for (auto& subgraph : subgraphs_) { in SetNumThreads() local
283 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32() local
292 for (auto& subgraph : subgraphs_) { in SetCancellationFunction() local
298 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegate() local
/external/dagger2/java/dagger/example/spi/
DBindingGraphVisualizer.java231 DotGraph subgraph = subgraph(component); in graph() local
249 DotGraph subgraph(ComponentPath component) { in subgraph() method in BindingGraphVisualizer.NodesGraph
250 DotGraph subgraph = new DotGraph("subgraph " + quote(clusterName(component))); in subgraph() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc58 std::unique_ptr<Graph> subgraph( in PartitionFunctionGraph() local
74 Graph* subgraph, const string& device_type, std::vector<int>* arg_indices, in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc617 const tflite::SubGraphT& subgraph, Location base_loc, in GetOutputTensorIndices()
656 const tflite::SubGraphT& subgraph, Builder* builder, const std::string name, in BuildTFEntryFunctionAttribute()
667 const tflite::SubGraphT& subgraph, ArrayRef<int32_t> output_indices) { in PruneSubgraph()
717 const tflite::SubGraphT& subgraph, llvm::StringRef name, in ConvertSubgraph()
921 std::string SubgraphName(unsigned index, const tflite::SubGraphT& subgraph) { in SubgraphName()
960 for (auto& subgraph : model->subgraphs) { in FlatBufferToMlir() local
983 auto& subgraph = e.value(); in FlatBufferToMlir() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_reader.cc40 const auto& subgraph = model->subgraphs[0]; in AddCalibrationToModel() local
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dhashtable_size.cc57 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableSize() local
Dhashtable_import.cc62 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableImport() local
Dhashtable_find.cc67 Subgraph* subgraph = reinterpret_cast<Subgraph*>(context->impl_); in EvalHashtableFind() local
/external/tensorflow/tensorflow/python/keras/utils/
Dvis_utils.py75 subgraph=False): argument

123