| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.cc | 49 std::unique_ptr<GPUOperation> SelectLSTM(const OperationDef& op_def, in SelectLSTM()
51 return absl::make_unique<GPUOperation>(CreateLSTM(op_def, gpu_info)); in SelectLSTM()
54 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr, in SelectReLU()
56 return absl::make_unique<GPUOperation>(CreateReLU(op_def, attr)); in SelectReLU()
59 std::unique_ptr<GPUOperation> SelectPReLU(const PReLUAttributes& attr, in SelectPReLU()
62 return absl::make_unique<GPUOperation>(CreatePReLU(gpu_info, op_def, attr)); in SelectPReLU()
65 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr, in SelectPooling()
67 return absl::make_unique<GPUOperation>(CreatePooling(op_def, attr)); in SelectPooling()
70 std::unique_ptr<GPUOperation> SelectMaxUnpooling( in SelectMaxUnpooling()
72 return absl::make_unique<GPUOperation>(CreateMaxUnpooling(op_def, attr)); in SelectMaxUnpooling()
[all …]
|
| D | simple_selectors.h | 30 std::unique_ptr<GPUOperation> SelectLSTM(const OperationDef& op_def,
33 std::unique_ptr<GPUOperation> SelectReLU(const ReLUAttributes& attr,
36 std::unique_ptr<GPUOperation> SelectPReLU(const PReLUAttributes& attr,
40 std::unique_ptr<GPUOperation> SelectPooling(const Pooling2DAttributes& attr,
43 std::unique_ptr<GPUOperation> SelectMaxUnpooling(
47 int dst_channels, std::unique_ptr<GPUOperation>* ptr);
51 std::unique_ptr<GPUOperation>* ptr);
56 std::unique_ptr<GPUOperation>* ptr);
58 std::unique_ptr<GPUOperation> SelectDWConvolutionDynamicWeights(
64 std::unique_ptr<GPUOperation>* ptr);
[all …]
|
| D | convolution_selector.h | 31 std::unique_ptr<GPUOperation> SelectConvolution(
35 std::unique_ptr<GPUOperation> SelectConvolutionForWinograd(
39 std::unique_ptr<GPUOperation> SelectConvolutionWithDynamicWeights(
44 std::unique_ptr<GPUOperation> SelectConverterToConvWeights(
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.h | 97 class GPUOperation {
99 GPUOperation() = default;
100 explicit GPUOperation(const OperationDef& definition);
101 virtual ~GPUOperation() = default;
103 GPUOperation(GPUOperation&& operation);
104 GPUOperation& operator=(GPUOperation&& operation);
105 GPUOperation(const GPUOperation&) = delete;
106 GPUOperation& operator=(const GPUOperation&) = delete;
108 absl::Status AddOperation(GPUOperation* operation);
157 friend flatbuffers::Offset<tflite::gpu::data::GPUOperation> Encode(
[all …]
|
| D | gpu_operation.cc | 75 GPUOperation::GPUOperation(const OperationDef& definition) in GPUOperation() function in tflite::gpu::GPUOperation
78 void GPUOperation::SetSrc(GpuSpatialTensor* ptr, int index) { in SetSrc()
85 void GPUOperation::SetDst(GpuSpatialTensor* ptr, int index) { in SetDst()
92 GPUOperation::GPUOperation(GPUOperation&& operation) in GPUOperation() function in tflite::gpu::GPUOperation
113 GPUOperation& GPUOperation::operator=(GPUOperation&& operation) { in operator =()
138 absl::Status GPUOperation::AddOperation(GPUOperation* operation) { in AddOperation()
158 void GPUOperation::AddSrcTensor(const std::string& tensor_name, in AddSrcTensor()
165 void GPUOperation::AddSrcBuffer(const std::string& buffer_name, in AddSrcBuffer()
172 void GPUOperation::AddSrcTexture2D(const std::string& texture_name, in AddSrcTexture2D()
179 void GPUOperation::AddDstTensor(const std::string& tensor_name, in AddDstTensor()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | elementwise_test_util.cc | 41 GPUOperation operation = CreateElementwiseOneInput( in AbsTest()
44 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AbsTest()
68 GPUOperation operation = CreateElementwiseOneInput( in CosTest()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in CosTest()
94 GPUOperation operation = CreateElementwiseOneInput( in CopyTest()
97 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in CopyTest()
119 GPUOperation operation = CreateElementwiseOneInput( in EluTest()
122 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in EluTest()
147 GPUOperation operation = CreateElementwiseOneInput( in ExpTest()
150 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ExpTest()
[all …]
|
| D | padding_test_util.cc | 46 GPUOperation operation = CreatePadding(op_def, attr); in PaddingAppendWidthTest()
48 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendWidthTest()
76 GPUOperation operation = CreatePadding(op_def, attr); in PaddingPrependWidthTest()
78 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingPrependWidthTest()
106 GPUOperation operation = CreatePadding(op_def, attr); in PaddingAppendHeightTest()
108 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendHeightTest()
135 GPUOperation operation = CreatePadding(op_def, attr); in PaddingPrependHeightTest()
137 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingPrependHeightTest()
164 GPUOperation operation = CreatePadding(op_def, attr); in PaddingAppendChannelsTest()
166 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PaddingAppendChannelsTest()
[all …]
|
| D | elementwise.cc | 184 GPUOperation CreateElementwiseOneRuntimeOneScalar( in CreateElementwiseOneRuntimeOneScalar()
187 GPUOperation op(definition); in CreateElementwiseOneRuntimeOneScalar()
202 GPUOperation CreateElementwiseTwoInput( in CreateElementwiseTwoInput()
214 GPUOperation result(definition); in CreateElementwiseTwoInput()
233 GPUOperation CreateElementwiseTwoInput( in CreateElementwiseTwoInput()
246 GPUOperation result(definition); in CreateElementwiseTwoInput()
268 GPUOperation CreateElementwiseOneInput(const GpuInfo& gpu_info, in CreateElementwiseOneInput()
271 GPUOperation op(definition); in CreateElementwiseOneInput()
278 GPUOperation CreateElementwise(const GpuInfo& gpu_info, in CreateElementwise()
299 return GPUOperation(definition); in CreateElementwise()
[all …]
|
| D | relu_test_util.cc | 46 GPUOperation operation = CreateReLU(op_def, attr); in ReLUNoClipNoAlphaTest()
48 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUNoClipNoAlphaTest()
75 GPUOperation operation = CreateReLU(op_def, attr); in ReLUClipTest()
77 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUClipTest()
104 GPUOperation operation = CreateReLU(op_def, attr); in ReLUAlphaTest()
106 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUAlphaTest()
133 GPUOperation operation = CreateReLU(op_def, attr); in ReLUAlphaClipTest()
135 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReLUAlphaClipTest()
|
| D | quantize_and_dequantize_test_util.cc | 51 GPUOperation operation = CreateQuantizeAndDequantize(op_def, attr); in QuantAndDequant_Dim2Bits8Test()
53 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim2Bits8Test()
88 GPUOperation operation = CreateQuantizeAndDequantize(op_def, attr); in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
90 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim3Bits8_NegativeRangeTest()
124 GPUOperation operation = CreateQuantizeAndDequantize(op_def, attr); in QuantAndDequant_Dim3Bits16Test()
126 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim3Bits16Test()
161 GPUOperation operation = CreateQuantizeAndDequantize(op_def, attr); in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
163 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in QuantAndDequant_Dim2Bits16_NegativeRangeTest()
|
| D | space_to_depth_test_util.cc | 42 GPUOperation operation = CreateSpaceToDepth(op_def, attr); in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
44 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x1BlockSize2Test()
69 GPUOperation operation = CreateSpaceToDepth(op_def, attr); in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x2BlockSize2Test()
98 GPUOperation operation = CreateSpaceToDepth(op_def, attr); in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
100 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x2x2x3BlockSize2Test()
129 GPUOperation operation = CreateSpaceToDepth(op_def, attr); in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
131 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in SpaceToDepthTensorShape1x4x4x1BlockSize2Test()
|
| D | pooling_test_util.cc | 49 GPUOperation operation = CreatePooling(op_def, attr); in AveragePoolingTest()
51 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingTest()
80 GPUOperation operation = CreatePooling(op_def, attr); in AveragePoolingNonEmptyPaddingTest()
82 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in AveragePoolingNonEmptyPaddingTest()
112 GPUOperation operation = CreatePooling(op_def, attr); in MaxPoolingTest()
114 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in MaxPoolingTest()
146 GPUOperation operation = CreatePooling(op_def, attr); in MaxPoolingIndicesTest()
148 {src_tensor}, absl::make_unique<GPUOperation>(std::move(operation)), in MaxPoolingIndicesTest()
|
| D | prelu_test_util.cc | 49 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUAlphaTest()
51 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaTest()
81 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUAlphaClipTest()
83 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PReLUAlphaClipTest()
113 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUHWCAlphaTest()
115 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in PReLUHWCAlphaTest()
|
| D | add_test_util.cc | 44 GPUOperation operation = CreateAdd(op_def, channels, channels[0]); in AddTwoEqualTensorsTest()
46 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in AddTwoEqualTensorsTest()
74 GPUOperation operation = CreateAdd(op_def, channels, channels[0]); in AddFirstTensorHasMoreChannelsThanSecondTest()
76 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in AddFirstTensorHasMoreChannelsThanSecondTest()
105 GPUOperation operation = CreateAdd(op_def, channels, 6); in AddFirstTensorHasLessChannelsThanSecond()
107 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in AddFirstTensorHasLessChannelsThanSecond()
|
| D | concat_test_util.cc | 49 GPUOperation operation = CreateConcatXY(op_def, attr); in ConcatWidthTest()
51 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in ConcatWidthTest()
82 GPUOperation operation = CreateConcatXY(op_def, attr); in ConcatHeightTest()
84 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in ConcatHeightTest()
117 GPUOperation operation = in ConcatChannelsTest()
121 absl::make_unique<GPUOperation>(std::move(operation)), in ConcatChannelsTest()
154 GPUOperation operation = CreateConcatZ(op_def, {4, 4}, env->GetGpuInfo()); in ConcatChannelsAlignedx4Test()
156 {src0, src1}, absl::make_unique<GPUOperation>(std::move(operation)), in ConcatChannelsAlignedx4Test()
|
| D | winograd.h | 31 class Winograd4x4To36 : public GPUOperation {
51 : GPUOperation(definition), padding_(padding) {} in Winograd4x4To36()
61 class Winograd4x4To36TileX6 : public GPUOperation {
98 class Winograd36To4x4 : public GPUOperation {
117 : GPUOperation(definition) {} in Winograd36To4x4()
127 class Winograd36To4x4Tile4x1 : public GPUOperation {
|
| D | depthwise_conv_test_util.cc | 52 GPUOperation operation = in DepthwiseConvSimpleWeightsTest()
55 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in DepthwiseConvSimpleWeightsTest()
89 GPUOperation operation = in DepthwiseConvNoMultiplierTest()
92 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in DepthwiseConvNoMultiplierTest()
127 GPUOperation operation = in DepthwiseConvMultiplier2Test()
130 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in DepthwiseConvMultiplier2Test()
|
| D | elementwise.h | 30 GPUOperation CreateElementwiseOneInput(const GpuInfo& gpu_info,
36 GPUOperation CreateElementwise(const GpuInfo& gpu_info,
43 GPUOperation CreateElementwiseTwoInput(const OperationDef& definition,
|
| D | reshape_test_util.cc | 43 GPUOperation operation = CreateReshape(op_def); in ReshapeTest()
45 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in ReshapeTest()
69 GPUOperation operation = CreateReshapex4(op_def); in Reshapex4Test()
71 src_tensor, absl::make_unique<GPUOperation>(std::move(operation)), in Reshapex4Test()
|
| D | max_unpooling.cc | 27 GPUOperation* op) { in GetMaxUnpoolingKernelCode()
128 GPUOperation CreateMaxUnpooling(const OperationDef& definition, in CreateMaxUnpooling()
130 GPUOperation op(definition); in CreateMaxUnpooling()
142 GPUOperation CreateMaxUnpooling(const OperationDef& definition, in CreateMaxUnpooling()
144 GPUOperation op(definition); in CreateMaxUnpooling()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/ |
| D | dw_convolution_selector.cc | 26 std::unique_ptr<GPUOperation> SelectDWConvolutionAdreno( in SelectDWConvolutionAdreno()
33 return absl::make_unique<GPUOperation>( in SelectDWConvolutionAdreno()
38 std::unique_ptr<GPUOperation> SelectDWConvolutionPowerVR( in SelectDWConvolutionPowerVR()
45 return absl::make_unique<GPUOperation>( in SelectDWConvolutionPowerVR()
50 std::unique_ptr<GPUOperation> SelectDWConvolutionMali( in SelectDWConvolutionMali()
62 return absl::make_unique<GPUOperation>( in SelectDWConvolutionMali()
67 std::unique_ptr<GPUOperation> SelectDWConvolutionApple( in SelectDWConvolutionApple()
77 return absl::make_unique<GPUOperation>( in SelectDWConvolutionApple()
83 std::unique_ptr<GPUOperation> SelectDWConvolution( in SelectDWConvolution()
|
| D | convolution_selector.cc | 36 std::unique_ptr<GPUOperation> SelectConvolutionAdreno( in SelectConvolutionAdreno()
41 GPUOperation conv = CreateConvConstants(gpu_info, op_def, attr); in SelectConvolutionAdreno()
42 return absl::make_unique<GPUOperation>(std::move(conv)); in SelectConvolutionAdreno()
49 std::unique_ptr<GPUOperation> SelectConvolutionWinogradAdreno( in SelectConvolutionWinogradAdreno()
58 std::unique_ptr<GPUOperation> SelectConvolutionDynamicWeightsAdreno( in SelectConvolutionDynamicWeightsAdreno()
69 std::unique_ptr<GPUOperation> SelectConvolutionNVidia( in SelectConvolutionNVidia()
73 GPUOperation conv = CreateConvConstants(gpu_info, op_def, attr); in SelectConvolutionNVidia()
74 return absl::make_unique<GPUOperation>(std::move(conv)); in SelectConvolutionNVidia()
81 std::unique_ptr<GPUOperation> SelectConvolutionPowerVR( in SelectConvolutionPowerVR()
88 std::unique_ptr<GPUOperation> SelectConvolutionMali( in SelectConvolutionMali()
[all …]
|
| D | fully_connected_selector.cc | 27 std::unique_ptr<GPUOperation> SelectFullyConnectedGeneric( in SelectFullyConnectedGeneric()
40 std::unique_ptr<GPUOperation> SelectFullyConnectedAdreno( in SelectFullyConnectedAdreno()
53 std::unique_ptr<GPUOperation> SelectFullyConnectedPowerVR( in SelectFullyConnectedPowerVR()
65 std::unique_ptr<GPUOperation> SelectFullyConnectedMali( in SelectFullyConnectedMali()
84 std::unique_ptr<GPUOperation> SelectFullyConnected( in SelectFullyConnected()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.h | 54 void Init(std::unique_ptr<GPUOperation>&& gpu_operation) { in Init()
58 GPUOperation& GetGpuOperation() { return *operation_; } in GetGpuOperation()
59 const GPUOperation& GetGpuOperation() const { return *operation_; } in GetGpuOperation()
90 std::unique_ptr<GPUOperation> operation_;
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.h | 55 std::unique_ptr<GPUOperation>&& operation,
85 std::unique_ptr<GPUOperation>&& operation,
90 std::unique_ptr<GPUOperation>&& operation,
95 std::unique_ptr<GPUOperation>&& operation,
|