Home
last modified time | relevance | path

Searched refs:graph_builder (Results 1 – 25 of 91) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_factory.h26 OpBuilder* CreateArgMinMaxOpBuilder(GraphBuilder* graph_builder, int op_type);
27 OpBuilder* CreateActivationBuilder(GraphBuilder* graph_builder, int op_type);
28 OpBuilder* CreateArithmeticBuilder(GraphBuilder* graph_builder, int op_type);
29 OpBuilder* CreateMatMulWithConstWeightsOpBuilder(GraphBuilder* graph_builder,
31 OpBuilder* CreateConcatBuilder(GraphBuilder* graph_builder, int op_type);
32 OpBuilder* CreateConv2DBuilder(GraphBuilder* graph_builder, int op_type);
33 OpBuilder* CreateTransposeConv2DBuilder(GraphBuilder* graph_builder,
35 OpBuilder* CreatePool2DBuilder(GraphBuilder* graph_builder, int op_type);
36 OpBuilder* CreateReshapeBuilder(GraphBuilder* graph_builder, int op_type);
37 OpBuilder* CreateSoftmaxBuilder(GraphBuilder* graph_builder, int op_type);
[all …]
Dmatmul_builder.cc48 GraphBuilder* graph_builder, in AddFullyConnectedHelper() argument
59 auto* data_min_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
61 auto* data_max_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
65 matmul_op->AddInput(graph_builder->GetHexagonTensorId(data_tensor_id)); in AddFullyConnectedHelper()
96 auto* bias_min_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
98 auto* bias_max_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
102 auto* bias_add_op = graph_builder->AddNode(matmul_op->GetTFLiteNodeID()); in AddFullyConnectedHelper()
105 bias_add_op->AddInput(graph_builder->GetHexagonTensorId(bias_tensor_id)); in AddFullyConnectedHelper()
124 auto* output_min_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
126 auto* output_max_const = graph_builder->AddConstNodeWithData( in AddFullyConnectedHelper()
[all …]
Dquantize_builder.h26 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type) in QuantizeOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in QuantizeOpBuilder()
28 explicit QuantizeOpBuilder(GraphBuilder* graph_builder, int op_type, in QuantizeOpBuilder() argument
30 : OpBuilder(graph_builder, op_type) {} in QuantizeOpBuilder()
Dactivation_builder.h28 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type) in ActivationOpBuilder() argument
29 : OpBuilder(graph_builder, op_type) {} in ActivationOpBuilder()
30 explicit ActivationOpBuilder(GraphBuilder* graph_builder, int op_type, in ActivationOpBuilder() argument
32 : OpBuilder(graph_builder, op_type), relu_value_(relu_value) {} in ActivationOpBuilder()
Dmatmul_builder.h29 explicit MatMulWithConstWeightsOpBuilder(GraphBuilder* graph_builder, in MatMulWithConstWeightsOpBuilder() argument
31 : OpBuilder(graph_builder, op_type) {} in MatMulWithConstWeightsOpBuilder()
49 explicit MatMulOpBuilder(GraphBuilder* graph_builder, int op_type) in MatMulOpBuilder() argument
50 : OpBuilder(graph_builder, op_type) {} in MatMulOpBuilder()
Dconv_2d_builder.h41 explicit Conv2dOpBuilder(GraphBuilder* graph_builder, int op_type) in Conv2dOpBuilder() argument
42 : OpBuilder(graph_builder, op_type) {} in Conv2dOpBuilder()
89 GraphBuilder* graph_builder, PerChannelQuantData* per_channel_quant);
94 GraphBuilder* graph_builder, PerChannelQuantData* per_channel_quant,
Dsquared_difference.cc23 explicit SquaredDifferenceOpBuilder(GraphBuilder* graph_builder, int op_type) in SquaredDifferenceOpBuilder() argument
24 : OpBuilder(graph_builder, op_type) {} in SquaredDifferenceOpBuilder()
97 OpBuilder* CreateSquaredDifferenceOpBuilder(GraphBuilder* graph_builder, in CreateSquaredDifferenceOpBuilder() argument
99 return new SquaredDifferenceOpBuilder(graph_builder, op_type); in CreateSquaredDifferenceOpBuilder()
Dtranspose_builder.h26 explicit TransposeOpBuilder(GraphBuilder* graph_builder, int op_type) in TransposeOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in TransposeOpBuilder()
Dneg_op_builder.h26 explicit NegOpBuilder(GraphBuilder* graph_builder, int op_type) in NegOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in NegOpBuilder()
Dstrided_slice_builder.h26 explicit StridedSliceOpBuilder(GraphBuilder* graph_builder, int op_type) in StridedSliceOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in StridedSliceOpBuilder()
Dslice_builder.h26 explicit SliceOpBuilder(GraphBuilder* graph_builder, int op_type) in SliceOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in SliceOpBuilder()
Dmin_max_builder.h26 explicit MinMaxOpBuilder(GraphBuilder* graph_builder, int op_type) in MinMaxOpBuilder() argument
27 : OpBuilder(graph_builder, op_type) {} in MinMaxOpBuilder()
Drsqrt_builder.cc27 explicit RsqrtOpBuilder(GraphBuilder* graph_builder, int op_type) in RsqrtOpBuilder() argument
28 : OpBuilder(graph_builder, op_type) {} in RsqrtOpBuilder()
166 OpBuilder* CreateRSqrtOpBuilder(GraphBuilder* graph_builder, int op_type) { in CreateRSqrtOpBuilder() argument
167 return new RsqrtOpBuilder(graph_builder, op_type); in CreateRSqrtOpBuilder()
Dcast_builder.h32 explicit CastOpBuilder(GraphBuilder* graph_builder, int op_type) in CastOpBuilder() argument
33 : OpBuilder(graph_builder, op_type) {} in CastOpBuilder()
Dsoftmax_builder.h28 explicit SoftmaxOpBuilder(GraphBuilder* graph_builder, int op_type) in SoftmaxOpBuilder() argument
29 : OpBuilder(graph_builder, op_type) {} in SoftmaxOpBuilder()
Dpad_builder.h28 explicit PadOpBuilder(GraphBuilder* graph_builder, int op_type) in PadOpBuilder() argument
29 : OpBuilder(graph_builder, op_type) {} in PadOpBuilder()
Dresize_nearest_neighbor_builder.h28 explicit ResizeNearestNeighborOpBuilder(GraphBuilder* graph_builder, in ResizeNearestNeighborOpBuilder() argument
30 : OpBuilder(graph_builder, op_type) {} in ResizeNearestNeighborOpBuilder()
Dl2_normalization_builder.h28 explicit L2NormalizationOpBuilder(GraphBuilder* graph_builder, int op_type) in L2NormalizationOpBuilder() argument
29 : OpBuilder(graph_builder, op_type) {} in L2NormalizationOpBuilder()
Dhardswish_builder.h30 explicit HardSwishOpBuilder(GraphBuilder* graph_builder, int op_type) in HardSwishOpBuilder() argument
31 : OpBuilder(graph_builder, op_type) {} in HardSwishOpBuilder()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dop_factory.h26 OpBuilder* CreateAddOpBuilder(GraphBuilder* graph_builder);
27 OpBuilder* CreateAveragePool2dOpBuilder(GraphBuilder* graph_builder);
28 OpBuilder* CreateConcatenationOpBuilder(GraphBuilder* graph_builder);
29 OpBuilder* CreateConvolutionOpBuilder(GraphBuilder* graph_builder);
30 OpBuilder* CreateDepthwiseConvolutionOpBuilder(GraphBuilder* graph_builder);
31 OpBuilder* CreateFullyConnectedOpBuilder(GraphBuilder* graph_builder);
32 OpBuilder* CreateHardSwishOpBuilder(GraphBuilder* graph_builder);
33 OpBuilder* CreateLogisticOpBuilder(GraphBuilder* graph_builder);
34 OpBuilder* CreateMaxPool2dOpBuilder(GraphBuilder* graph_builder);
35 OpBuilder* CreateMeanOpBuilder(GraphBuilder* graph_builder);
[all …]
Dactivation_layer_builder.cc114 OpBuilder* CreateActivationLayerBuilder(GraphBuilder* graph_builder) { in CreateActivationLayerBuilder() argument
115 return new ActivationLayerBuilder(graph_builder); in CreateActivationLayerBuilder()
118 OpBuilder* CreateLogisticOpBuilder(GraphBuilder* graph_builder) { in CreateLogisticOpBuilder() argument
119 return new ActivationLayerBuilder(graph_builder, kTfLiteActSigmoid); in CreateLogisticOpBuilder()
122 OpBuilder* CreateReluOpBuilder(GraphBuilder* graph_builder) { in CreateReluOpBuilder() argument
123 return new ActivationLayerBuilder(graph_builder, kTfLiteActRelu); in CreateReluOpBuilder()
126 OpBuilder* CreateReluN1To1OpBuilder(GraphBuilder* graph_builder) { in CreateReluN1To1OpBuilder() argument
127 return new ActivationLayerBuilder(graph_builder, kTfLiteActReluN1To1); in CreateReluN1To1OpBuilder()
130 OpBuilder* CreateRelu6OpBuilder(GraphBuilder* graph_builder) { in CreateRelu6OpBuilder() argument
131 return new ActivationLayerBuilder(graph_builder, kTfLiteActRelu6); in CreateRelu6OpBuilder()
[all …]
Dpooling_layer_builder.cc123 OpBuilder* CreateAveragePool2dOpBuilder(GraphBuilder* graph_builder) { in CreateAveragePool2dOpBuilder() argument
124 return new PoolingLayerBuilder(graph_builder, kTfLiteBuiltinAveragePool2d); in CreateAveragePool2dOpBuilder()
127 OpBuilder* CreateMaxPool2dOpBuilder(GraphBuilder* graph_builder) { in CreateMaxPool2dOpBuilder() argument
128 return new PoolingLayerBuilder(graph_builder, kTfLiteBuiltinMaxPool2d); in CreateMaxPool2dOpBuilder()
131 OpBuilder* CreateMeanOpBuilder(GraphBuilder* graph_builder) { in CreateMeanOpBuilder() argument
132 return new PoolingLayerBuilder(graph_builder, kTfLiteBuiltinMean); in CreateMeanOpBuilder()
Dactivation_layer_builder.h27 explicit ActivationLayerBuilder(GraphBuilder* graph_builder) in ActivationLayerBuilder() argument
28 : OpBuilder(graph_builder) {} in ActivationLayerBuilder()
30 explicit ActivationLayerBuilder(GraphBuilder* graph_builder, in ActivationLayerBuilder() argument
32 : OpBuilder(graph_builder), activation_(activation) {} in ActivationLayerBuilder()
Dpad_op_builder.cc89 OpBuilder* CreatePadOpBuilder(GraphBuilder* graph_builder) { in CreatePadOpBuilder() argument
90 return new PadOpBuilder(graph_builder, PadType::kPad); in CreatePadOpBuilder()
93 OpBuilder* CreateMirrorPadOpBuilder(GraphBuilder* graph_builder) { in CreateMirrorPadOpBuilder() argument
94 return new PadOpBuilder(graph_builder, PadType::kMirrorPad); in CreateMirrorPadOpBuilder()
Dconvolution_op_builder.cc285 OpBuilder* CreateConvolutionOpBuilder(GraphBuilder* graph_builder) { in CreateConvolutionOpBuilder() argument
286 return new ConvolutionOpBuilder(graph_builder, ConvolutionType::kConv); in CreateConvolutionOpBuilder()
289 OpBuilder* CreateDepthwiseConvolutionOpBuilder(GraphBuilder* graph_builder) { in CreateDepthwiseConvolutionOpBuilder() argument
290 return new ConvolutionOpBuilder(graph_builder, in CreateDepthwiseConvolutionOpBuilder()
294 OpBuilder* CreateTransposeConvolutionOpBuilder(GraphBuilder* graph_builder) { in CreateTransposeConvolutionOpBuilder() argument
295 return new ConvolutionOpBuilder(graph_builder, in CreateTransposeConvolutionOpBuilder()

1234