Home
last modified time | relevance | path

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

123456

/external/XNNPACK/test/
Dmemory-planner-test.cc17 struct xnn_subgraph subgraph; in TEST() local
54 struct xnn_subgraph subgraph; in TEST() local
103 struct xnn_subgraph subgraph; in TEST() local
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc81 void SetupTensor(Subgraph* subgraph, int tensor_index, TfLiteType type) { in SetupTensor()
95 void SubgraphBuilder::BuildAddSubgraph(Subgraph* subgraph) { in BuildAddSubgraph()
126 void SubgraphBuilder::BuildMulSubgraph(Subgraph* subgraph) { in BuildMulSubgraph()
157 void SubgraphBuilder::BuildPadSubgraph(Subgraph* subgraph) { in BuildPadSubgraph()
186 void SubgraphBuilder::BuildIfSubgraph(Subgraph* subgraph) { in BuildIfSubgraph()
221 void SubgraphBuilder::BuildLessEqualCondSubgraph(Subgraph* subgraph, int rhs) { in BuildLessEqualCondSubgraph()
254 void SubgraphBuilder::BuildAccumulateLoopBodySubgraph(Subgraph* subgraph) { in BuildAccumulateLoopBodySubgraph()
300 void SubgraphBuilder::BuildPadLoopBodySubgraph(Subgraph* subgraph, in BuildPadLoopBodySubgraph()
354 void SubgraphBuilder::BuildWhileSubgraph(Subgraph* subgraph) { in BuildWhileSubgraph()
390 Subgraph* subgraph) { in BuildAssignRandomValueToVariableSubgraph()
[all …]
/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()
99 struct xnn_node* xnn_subgraph_new_node(xnn_subgraph_t subgraph) in xnn_subgraph_new_node()
130 uint32_t xnn_check_nchw_compatibility(xnn_subgraph_t subgraph, struct xnn_node* node) { in xnn_check_nchw_compatibility()
272 void xnn_subgraph_rewrite_for_nchw(xnn_subgraph_t subgraph) in xnn_subgraph_rewrite_for_nchw()
490 xnn_subgraph_t subgraph, in xnn_subgraph_optimize()
675 xnn_subgraph_t subgraph) in xnn_delete_subgraph()
Dtensor.c19 xnn_subgraph_t subgraph, in xnn_define_tensor_value()
75 xnn_subgraph_t subgraph, in xnn_tensor_get_size()
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc33 inline int GetNumDims(const SubGraph* subgraph, const Operator* op, int idx) { in GetNumDims()
39 inline bool HaveSameShapes(const SubGraph* subgraph, const Operator* op, in HaveSameShapes()
655 TensorType GetTensorType(int32_t idx, const SubGraph* subgraph) { in GetTensorType()
675 const SubGraph* subgraph) { in GetOpSignature()
854 const SubGraph* subgraph = subgraphs->Get(i); in UpdateOpVersion() local
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc136 std::unique_ptr<xnn_subgraph, decltype(&xnn_delete_subgraph)> subgraph( in Create() local
813 xnn_subgraph_t subgraph, TfLiteContext* context, in VisitNode()
1041 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitAbsNode()
1074 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitAddNode()
1124 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitAveragePool2DNode()
1190 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitCeilNode()
1223 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitConv2DNode()
1321 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitDepthwiseConv2DNode()
1422 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitDepthToSpaceNode()
1468 xnn_subgraph_t subgraph, TfLiteContext* logging_context, int node_index, in VisitDivNode()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model_test.cc241 for (const auto& subgraph : model_.subgraphs) { in TEST_P() local
263 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_P() local
328 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_P() local
442 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_P() local
544 const auto& subgraph = model_.subgraphs[subgraph_idx]; in TEST_F() local
607 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
717 const auto& subgraph = model_.subgraphs[0]; in TEST_P() local
831 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
893 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
956 const auto& subgraph = model_.subgraphs[0]; in TEST_F() local
[all …]
Dquantize_model.cc42 bool IsFloatTensor(const SubGraphT* subgraph, int32_t tensor_idx) { in IsFloatTensor()
60 const SubGraphT* subgraph = model->subgraphs[subgraph_index].get(); in GetOperatorProperty() local
91 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in PopulateRealValueOpSet() local
223 bool InputQuantizeRequired(const ModelT* model, const SubGraphT* subgraph, in InputQuantizeRequired()
259 int32_t SetInputType(ModelT* model, SubGraphT* subgraph, in SetInputType()
323 int32_t SetOutputType(ModelT* model, SubGraphT* subgraph, in SetOutputType()
392 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetInputAndOutputTypes() local
444 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in ApplyConstraints() local
526 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetOperatorPropertyADDSUBOperator() local
585 bool IsSubgraphInput(SubGraphT* subgraph, int32_t index) { in IsSubgraphInput()
[all …]
Dmodify_model_interface.cc55 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetInputTensors() local
130 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetOutputTensors() local
200 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetInputTypeToUINT8() local
227 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetOutputTypeToUINT8() local
260 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveInputTensor() local
286 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveOutputTensor() local
398 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddUint8Dequant() local
438 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddUint8Quant() local
Dquantization_utils_test.cc473 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
523 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
558 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
581 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
627 auto subgraph = model.subgraphs[0].get(); in TEST_F() local
650 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
683 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
725 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
762 auto subgraph = absl::make_unique<SubGraphT>(); in TEST_F() local
783 auto subgraph = absl::make_unique<tflite::SubGraphT>(); in TEST_F() local
Dquantize_weights.cc59 const SubGraphT* subgraph, in GetTensorConsumers()
171 bool CheckAllOpInputsQuantized(const SubGraphT* subgraph, const OperatorT* op, in CheckAllOpInputsQuantized()
200 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in InsertQuantizableInputTensorsFromOperator() local
389 const ModelT* model, const SubGraphT* subgraph, in PassQuantizationAndGetConsumers()
432 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in QuantizeWeightsInt8() local
556 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in QuantizeWeightsFloat16() local
Dquantization_wrapper_utils.cc50 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in IntermediateTensorExists() local
88 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddIntermediateTensorsToFusedOp() local
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator.cc98 const SubGraph* subgraph = (*subgraphs)[0]; in CheckOfflinePlannedOffsets() local
190 TfLiteStatus AllocationInfoBuilder::AddTensors(const SubGraph* subgraph, in AddTensors()
662 const SubGraph* subgraph = GetSubGraphFromModel(model); in FinishModelAllocation() local
757 const SubGraph* subgraph = GetSubGraphFromModel(model); in AllocateNodeAndRegistrations() local
780 const SubGraph* subgraph = GetSubGraphFromModel(model); in PrepareNodeAndRegistrationDataFromFlatbuffer() local
868 const SubGraph* subgraph = GetSubGraphFromModel(model); in AllocatePersistentTfLiteTensor() local
900 const SubGraph* subgraph = GetSubGraphFromModel(model); in AllocateTempTfLiteTensor() local
939 const SubGraph* subgraph = GetSubGraphFromModel(model); in AllocateTfLiteEvalTensors() local
968 TfLiteStatus MicroAllocator::AllocateVariables(const SubGraph* subgraph, in AllocateVariables()
998 const Model* model, const SubGraph* subgraph, TfLiteTensor* tensor, in PopulateTfLiteTensorFromFlatbuffer()
[all …]
/external/tensorflow/tensorflow/lite/
Dinterpreter.cc249 Subgraph* subgraph = new Subgraph(error_reporter_, external_contexts_, in AddSubgraphs() local
357 for (auto& subgraph : subgraphs_) { in SetNumThreads() local
371 for (auto& subgraph : subgraphs_) { in SetAllowFp16PrecisionForFp32() local
380 for (auto& subgraph : subgraphs_) { in SetCancellationFunction() local
389 for (auto& subgraph : subgraphs_) { in ModifyGraphWithDelegate() local
407 for (auto& subgraph : subgraphs_) { in RemoveAllDelegates() local
/external/XNNPACK/src/subgraph/
Dfloor.c17 xnn_subgraph_t subgraph, in xnn_define_floor()
Dabs.c17 xnn_subgraph_t subgraph, in xnn_define_abs()
Dhardswish.c17 xnn_subgraph_t subgraph, in xnn_define_hardswish()
Dbankers-rounding.c17 xnn_subgraph_t subgraph, in xnn_define_bankers_rounding()
Dsquare-root.c17 xnn_subgraph_t subgraph, in xnn_define_square_root()
Dsigmoid.c17 xnn_subgraph_t subgraph, in xnn_define_sigmoid()
Dsquare.c17 xnn_subgraph_t subgraph, in xnn_define_square()
Dnegate.c17 xnn_subgraph_t subgraph, in xnn_define_negate()
Dsoftmax.c17 xnn_subgraph_t subgraph, in xnn_define_softmax()
Dceiling.c17 xnn_subgraph_t subgraph, in xnn_define_ceiling()
/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

123456