Home
last modified time | relevance | path

Searched refs:OperationDef (Results 1 – 25 of 138) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dsimple_selectors.h30 std::unique_ptr<GPUOperation> SelectLSTM(const OperationDef& op_def,
34 const OperationDef& op_def);
38 const OperationDef& op_def);
41 const OperationDef& op_def);
44 const MaxUnpooling2DAttributes& attr, const OperationDef& op_def);
46 void SelectAdd(const OperationDef& op_def, const std::vector<int>& channels,
50 const OperationDef& op_def,
55 const OperationDef& op_def, const GpuInfo& gpu_info,
60 const OperationDef& op_def);
63 const OperationDef& op_def,
[all …]
Dsimple_selectors.cc49 std::unique_ptr<GPUOperation> SelectLSTM(const OperationDef& op_def, in SelectLSTM()
55 const OperationDef& op_def) { in SelectReLU()
61 const OperationDef& op_def) { in SelectPReLU()
66 const OperationDef& op_def) { in SelectPooling()
71 const MaxUnpooling2DAttributes& attr, const OperationDef& op_def) { in SelectMaxUnpooling()
75 void SelectAdd(const OperationDef& op_def, const std::vector<int>& channels, in SelectAdd()
82 const OperationDef& op_def, in SelectResize()
91 const OperationDef& op_def, const GpuInfo& gpu_info, in SelectConcat()
114 const OperationDef& op_def) { in SelectDWConvolutionDynamicWeights()
120 const OperationDef& op_def, in SelectReshape()
[all …]
Dconvolution_selector.h33 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
37 const GpuInfo& gpu_info, const OperationDef& op_def, ModelHints hints);
41 const BHWC& dst_shape, const GpuInfo& gpu_info, const OperationDef& op_def,
45 const WeightsDescription& weights_desc, const OperationDef& op_def,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h50 Winograd4x4To36(const OperationDef& definition, const Padding2D& padding) in Winograd4x4To36()
52 friend Winograd4x4To36 CreateWinograd4x4To36(const OperationDef& definition,
58 Winograd4x4To36 CreateWinograd4x4To36(const OperationDef& definition,
64 Winograd4x4To36TileX6(const OperationDef& definition,
81 const GpuInfo& gpu_info, const OperationDef& definition,
86 std::string GetWinograd4x4To36TileX6Code(const OperationDef& op_def);
95 const GpuInfo& gpu_info, const OperationDef& definition,
116 explicit Winograd36To4x4(const OperationDef& definition) in Winograd36To4x4()
119 const OperationDef& definition,
124 const OperationDef& definition,
[all …]
Dresize.h38 friend Resize CreateResize(const OperationDef& definition,
42 Resize(const OperationDef& definition, const Resize2DAttributes& attr);
44 std::string GetResizeCode(const OperationDef& op_def,
50 Resize CreateResize(const OperationDef& definition,
64 friend Resize3D CreateResize3D(const OperationDef& definition,
68 Resize3D(const OperationDef& definition, const Resize3DAttributes& attr);
70 std::string GetResize3DCode(const OperationDef& op_def,
76 Resize3D CreateResize3D(const OperationDef& definition,
Dconv_powervr.h110 ConvPowerVR(const OperationDef& definition,
113 ConvPowerVR(const OperationDef& definition,
116 ConvPowerVR(const OperationDef& definition,
119 explicit ConvPowerVR(const OperationDef& definition);
120 ConvPowerVR(const OperationDef& definition,
143 const OperationDef& definition,
148 const OperationDef& definition,
153 const GpuInfo& gpu_info, const OperationDef& definition,
158 const GpuInfo& gpu_info, const OperationDef& definition,
162 const OperationDef& definition,
[all …]
Dconvolution_transposed_3x3.h69 ConvolutionTransposed3x3(const OperationDef& definition,
72 const GpuInfo& gpu_info, const OperationDef& definition,
75 const GpuInfo& gpu_info, const OperationDef& definition,
84 const GpuInfo& gpu_info, const OperationDef& op_def,
94 const OperationDef& definition,
98 const GpuInfo& gpu_info, const OperationDef& definition,
102 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_4x4.h71 ConvolutionTransposed4x4(const OperationDef& definition,
75 const GpuInfo& gpu_info, const OperationDef& definition,
78 const GpuInfo& gpu_info, const OperationDef& definition,
88 const GpuInfo& gpu_info, const OperationDef& op_def,
95 const OperationDef& definition,
99 const GpuInfo& gpu_info, const OperationDef& definition,
103 const GpuInfo& gpu_info, const OperationDef& definition,
Dconvolution_transposed_3x3_thin.h60 const OperationDef& definition,
64 const GpuInfo& gpu_info, const OperationDef& definition,
68 const GpuInfo& gpu_info, const OperationDef& definition,
76 std::string GenerateConvolutionTransposedCode(const OperationDef& op_def,
86 const GpuInfo& gpu_info, const OperationDef& definition,
90 const GpuInfo& gpu_info, const OperationDef& definition,
Dconv_buffer_1x1.h69 ConvBuffer1x1(const OperationDef& definition, const ConvParams& conv_params);
71 const OperationDef& definition,
75 const OperationDef& definition,
79 const GpuInfo& gpu_info, const OperationDef& definition,
82 const GpuInfo& gpu_info, const OperationDef& definition,
100 const OperationDef& op_def, const ConvBuffer1x1::ConvParams& conv_params,
169 bool IsConvBuffer1x1Supported(const OperationDef& definition,
172 bool IsConvBuffer1x1Supported(const OperationDef& definition,
177 const OperationDef& definition,
182 const OperationDef& definition,
[all …]
Dconv_metal.h80 explicit ConvolutionMetal(const OperationDef& definition) in ConvolutionMetal()
83 const OperationDef& definition, const BHWC& dst_shape,
87 const OperationDef& definition, const BHWC& dst_shape,
95 ConvolutionMetal CreateConvolutionMetal(const OperationDef& definition,
101 const OperationDef& definition, const BHWC& dst_shape,
104 bool IsConvolutionMetalSupported(const OperationDef& definition);
Dconvolution_transposed.h64 const GpuInfo& gpu_info, const OperationDef& definition,
67 const GpuInfo& gpu_info, const OperationDef& definition,
70 const GpuInfo& gpu_info, const OperationDef& definition,
73 ConvolutionTransposed(const OperationDef& definition,
76 ConvolutionTransposed(const OperationDef& definition,
88 std::string GenerateConvolutionTransposedCode(const OperationDef& op_def,
203 const GpuInfo& gpu_info, const OperationDef& definition,
207 const GpuInfo& gpu_info, const OperationDef& definition,
211 const GpuInfo& gpu_info, const OperationDef& definition,
Dreduce.h34 const OperationDef& definition, const GpuInfo& gpu_info);
50 std::string GetReduceKernelCode(const OperationDef& op_def,
59 OperationType op_type, const OperationDef& definition,
64 const OperationDef& definition, const GpuInfo& gpu_info);
Delementwise_test_util.cc35 OperationDef op_def; in AbsTest()
62 OperationDef op_def; in CosTest()
88 OperationDef op_def; in CopyTest()
113 OperationDef op_def; in EluTest()
141 OperationDef op_def; in ExpTest()
169 OperationDef op_def; in HardSwishTest()
196 OperationDef op_def; in LogTest()
223 OperationDef op_def; in NegTest()
249 OperationDef op_def; in RsqrtTest()
278 OperationDef op_def; in SigmoidTest()
[all …]
Delementwise.h31 const OperationDef& definition,
37 const OperationDef& definition,
43 GPUOperation CreateElementwiseTwoInput(const OperationDef& definition,
Dstrided_slice.h28 StridedSlice(const OperationDef& definition, const SliceAttributes& attr);
39 std::string GetStridedSliceCode(const OperationDef& op_def, bool alignedx4);
44 StridedSlice CreateStridedSlice(const OperationDef& definition,
Dconv_weights_converter.h29 ConverterToConvWeights(const OperationDef& definition,
42 const OperationDef& op_def, const WeightsDescription& weights_desc);
52 const OperationDef& definition, const WeightsDescription& weights_desc);
Dpadding_test_util.cc40 OperationDef op_def; in PaddingAppendWidthTest()
70 OperationDef op_def; in PaddingPrependWidthTest()
100 OperationDef op_def; in PaddingAppendHeightTest()
129 OperationDef op_def; in PaddingPrependHeightTest()
158 OperationDef op_def; in PaddingAppendChannelsTest()
187 OperationDef op_def; in PaddingPrependChannelsTest()
216 OperationDef op_def; in PaddingPrependChannelsX4Test()
245 OperationDef op_def; in PaddingComplexTest()
278 OperationDef op_def; in PaddingReflectWidthTest()
308 OperationDef op_def; in PaddingReflectChannelsTest()
Dsoftmax1x1.h28 explicit Softmax1x1(const OperationDef& definition);
47 std::string GetSoftmaxKernelCode(const OperationDef& op_def);
50 Softmax1x1 CreateSoftmax1x1(const OperationDef& definition);
Dresize_test_util.cc41 OperationDef op_def; in ResizeBilinearAlignedTest()
73 OperationDef op_def; in ResizeBilinearNonAlignedTest()
106 OperationDef op_def; in ResizeBilinearWithoutHalfPixelTest()
138 OperationDef op_def; in ResizeBilinearWithHalfPixelTest()
170 OperationDef op_def; in ResizeNearestTest()
202 OperationDef op_def; in ResizeNearestAlignCornersTest()
234 OperationDef op_def; in ResizeNearestHalfPixelCentersTest()
Dresize.cc24 Resize::Resize(const OperationDef& definition, const Resize2DAttributes& attr) in Resize()
40 std::string Resize::GetResizeCode(const OperationDef& op_def, in GetResizeCode()
153 Resize CreateResize(const OperationDef& definition, in CreateResize()
158 Resize3D::Resize3D(const OperationDef& definition, in Resize3D()
175 std::string Resize3D::GetResize3DCode(const OperationDef& op_def, in GetResize3DCode()
310 Resize3D CreateResize3D(const OperationDef& definition, in CreateResize3D()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc38 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionAdreno()
51 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionWinogradAdreno()
61 const OperationDef& op_def, ModelHints hints, in SelectConvolutionDynamicWeightsAdreno()
71 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionNVidia()
83 const OperationDef& op_def) { in SelectConvolutionPowerVR()
90 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionMali()
104 const GpuInfo& gpu_info, const OperationDef& op_def) { in SelectConvolutionWinogradMali()
119 const OperationDef& op_def, ModelHints hints, in SelectConvolutionDynamicWeightsMali()
139 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolution()
161 const GpuInfo& gpu_info, const OperationDef& op_def, in SelectConvolutionForWinograd()
[all …]
Ddw_convolution_selector.cc28 const OperationDef& op_def) { in SelectDWConvolutionAdreno()
40 const OperationDef& op_def) { in SelectDWConvolutionPowerVR()
52 const OperationDef& op_def) { in SelectDWConvolutionMali()
69 const OperationDef& op_def) { in SelectDWConvolutionApple()
85 const OperationDef& op_def) { in SelectDWConvolution()
Dfully_connected_selector.cc29 const OperationDef& op_def, int batch_size) { in SelectFullyConnectedGeneric()
42 const OperationDef& op_def, int batch_size) { in SelectFullyConnectedAdreno()
55 const OperationDef& op_def, int batch_size) { in SelectFullyConnectedPowerVR()
67 const OperationDef& op_def, int batch_size) { in SelectFullyConnectedMali()
86 const OperationDef& op_def, int batch_size) { in SelectFullyConnected()
Dconvolution_transposed_selector.cc30 const OperationDef& op_def) { in SelectConvolutionTransposedAdreno()
48 const OperationDef& op_def) { in SelectConvolutionTransposedPowerVR()
74 const OperationDef& op_def) { in SelectConvolutionTransposedMali()
83 const OperationDef& op_def) { in SelectConvolutionTransposed()
98 const OperationDef& op_def, WeightsDescription* weights_desc) { in SelectConvolutionTransposedWithDynamicWeights()

123456