Searched refs:AddBuilder (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.cc | 36 OpBuilder* GraphBuilder::AddBuilder(int builtin_code, const TfLiteNode* node) { in AddBuilder() function in tflite::delegates::coreml::GraphBuilder
39 return AddBuilder(CreateAddOpBuilder, node); in AddBuilder()
41 return AddBuilder(CreateAveragePool2dOpBuilder, node); in AddBuilder()
43 return AddBuilder(CreateConcatenationOpBuilder, node); in AddBuilder()
45 return AddBuilder(CreateConvolutionOpBuilder, node); in AddBuilder()
47 return AddBuilder(CreateDepthwiseConvolutionOpBuilder, node); in AddBuilder()
52 return AddBuilder(CreateDummyOpBuilder, node); in AddBuilder()
54 return AddBuilder(CreateFullyConnectedOpBuilder, node); in AddBuilder()
56 return AddBuilder(CreateLogisticOpBuilder, node); in AddBuilder()
58 return AddBuilder(CreateMaxPool2dOpBuilder, node); in AddBuilder()
[all …]
|
| D | hardswish_op_builder.cc | 42 graph_builder_->AddBuilder(CreateAddOpBuilder, nullptr)); in PopulateSubgraph()
51 graph_builder_->AddBuilder(CreateMulOpBuilder, nullptr)); in PopulateSubgraph()
|
| D | op_builder.h | 59 OpBuilder* AddBuilder(int builtin_code, const TfLiteNode* node);
62 OpBuilder* AddBuilder(const std::function<OpBuilder*(GraphBuilder*)>& builder,
|
| D | activation_layer_builder.cc | 76 graph_builder_->AddBuilder(CreateThresholdLayerBuilder, nullptr)); in PopulateSubgraph()
85 graph_builder_->AddBuilder(CreateActivationLayerBuilder, nullptr)); in PopulateSubgraph()
|
| D | add_op_builder.cc | 54 graph_builder_->AddBuilder(CreateActivationLayerBuilder, nullptr)); in PopulateSubgraph()
|
| D | mul_op_builder.cc | 60 graph_builder_->AddBuilder(CreateActivationLayerBuilder, nullptr)); in PopulateSubgraph()
|
| D | fully_connected_op_builder.cc | 97 graph_builder_->AddBuilder(CreateActivationLayerBuilder, nullptr)); in PopulateSubgraph()
|
| D | convolution_op_builder.cc | 233 graph_builder_->AddBuilder(CreateActivationLayerBuilder, nullptr)); in PopulateSubgraph()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 126 auto* op_builder = builder_->AddBuilder(reg->builtin_code, node);
|