| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | fully_connected_test_util.cc | 50 CreateFullyConnected(env->GetGpuInfo(), op_def, attr); in FullyConnectedTest()
96 CreateFullyConnected(env->GetGpuInfo(), op_def, attr); in FullyConnectedLargeTest()
139 if (!env->GetGpuInfo().IsRoundToNearestSupported()) { in FullyConnectedExtraLargeTest()
143 env->GetGpuInfo().IsApiMetal() && env->GetGpuInfo().IsIntel()) { in FullyConnectedExtraLargeTest()
153 CreateFullyConnected(env->GetGpuInfo(), op_def, attr); in FullyConnectedExtraLargeTest()
|
| D | elementwise_test_util.cc | 42 env->GetGpuInfo(), op_def, OperationType::ABS); in AbsTest()
69 env->GetGpuInfo(), op_def, OperationType::COS); in CosTest()
95 env->GetGpuInfo(), op_def, OperationType::COPY); in CopyTest()
120 env->GetGpuInfo(), op_def, OperationType::ELU); in EluTest()
148 env->GetGpuInfo(), op_def, OperationType::EXP); in ExpTest()
176 env->GetGpuInfo(), op_def, OperationType::HARD_SWISH); in HardSwishTest()
203 env->GetGpuInfo(), op_def, OperationType::LOG); in LogTest()
230 env->GetGpuInfo(), op_def, OperationType::NEG); in NegTest()
256 env->GetGpuInfo(), op_def, OperationType::RSQRT); in RsqrtTest()
285 env->GetGpuInfo(), op_def, OperationType::SIGMOID); in SigmoidTest()
[all …]
|
| D | reduce_test_util.cc | 45 env->GetGpuInfo()); in MeanHWTest()
72 op_def, env->GetGpuInfo()); in ReduceSumChannelsTest()
99 op_def, env->GetGpuInfo()); in ReduceProductChannelsTest()
127 op_def, env->GetGpuInfo()); in ReduceMaxChannelsTest()
155 op_def, env->GetGpuInfo()); in ReduceMinChannelsTest()
|
| D | winograd_test_util.cc | 71 if (!env->GetGpuInfo().IsRoundToNearestSupported()) { in Winograd4x4To36TileX6Test()
84 CreateWinograd4x4To36TileX6(env->GetGpuInfo(), op_def, padding); in Winograd4x4To36TileX6Test()
143 if (!env->GetGpuInfo().IsRoundToNearestSupported()) { in Winograd36To4x4Tile4x1Test()
153 CreateWinograd36To4x4Tile4x1(env->GetGpuInfo(), op_def, biases); in Winograd36To4x4Tile4x1Test()
207 if (!env->GetGpuInfo().IsRoundToNearestSupported()) { in Winograd4x4To36Test()
278 if (!env->GetGpuInfo().IsRoundToNearestSupported()) { in Winograd36To4x4Test()
|
| D | conv_powervr_test_util.cc | 53 CreateConvPowerVR(env->GetGpuInfo(), op_def, attr); in ConvPowerVR1x1SimpleWeightsTest()
90 CreateConvPowerVR(env->GetGpuInfo(), op_def, attr); in ConvPowerVR1x1Test()
127 CreateConvPowerVR(env->GetGpuInfo(), op_def, attr); in ConvPowerVRSimpleWeightsTest()
164 CreateConvPowerVR(env->GetGpuInfo(), op_def, attr); in ConvPowerVRTest()
|
| D | prelu_test_util.cc | 49 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUAlphaTest()
81 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUAlphaClipTest()
113 GPUOperation operation = CreatePReLU(env->GetGpuInfo(), op_def, attr); in PReLUHWCAlphaTest()
|
| D | depthwise_conv_test_util.cc | 53 CreateDepthwiseConvolution2D(env->GetGpuInfo(), op_def, attr); in DepthwiseConvSimpleWeightsTest()
90 CreateDepthwiseConvolution2D(env->GetGpuInfo(), op_def, attr); in DepthwiseConvNoMultiplierTest()
128 CreateDepthwiseConvolution2D(env->GetGpuInfo(), op_def, attr); in DepthwiseConvMultiplier2Test()
|
| D | mean_stddev_normalization_test_util.cc | 46 CreateMeanStdDevNormalization(op_def, env->GetGpuInfo(), 1); in MeanStddevNormSeparateBatchesTest()
93 CreateMeanStdDevNormalization(op_def, env->GetGpuInfo(), 1); in MeanStddevNormalizationAllBatchesTest()
147 auto operation = CreateMeanStdDevNormalization(op_def, env->GetGpuInfo(), in MeanStddevNormalizationLargeVectorTest()
|
| D | convolution_transposed_test_util.cc | 54 CreateConvolutionTransposed(env->GetGpuInfo(), op_def, attr); in ConvolutionTransposedSimpleWeightsTest()
94 CreateConvolutionTransposed(env->GetGpuInfo(), op_def, attr); in ConvolutionTransposedTest()
|
| D | depthwise_conv_3x3_test_util.cc | 54 CreateDepthwiseConv3x3(env->GetGpuInfo(), op_def, attr); in DepthwiseConv3x3SimpleWeightsTest()
92 CreateDepthwiseConv3x3(env->GetGpuInfo(), op_def, attr); in DepthwiseConv3x3Test()
|
| D | convolution_transposed_4x4_test_util.cc | 54 CreateConvolutionTransposed4x4(env->GetGpuInfo(), op_def, attr); in ConvolutionTransposed4x4SimpleWeightsTest()
|
| D | depthwise_conv_3x3_stride_h2_test_util.cc | 54 CreateDepthWiseConv3x3StrideH2(op_def, attr, env->GetGpuInfo()); in DepthWiseConv3x3StrideH2SimpleWeightsTest()
|
| D | concat_test_util.cc | 118 CreateConcatZ(op_def, {1, 2, 3}, env->GetGpuInfo()); in ConcatChannelsTest()
154 GPUOperation operation = CreateConcatZ(op_def, {4, 4}, env->GetGpuInfo()); in ConcatChannelsAlignedx4Test()
|
| D | lstm_test_util.cc | 60 GPUOperation operation = CreateLSTM(op_def, env->GetGpuInfo()); in LstmTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.cc | 184 operation_->AssembleCode(creation_context.GetGpuInfo()); in Compile()
189 creation_context.GetGpuInfo(), in Compile()
195 return operation_->PostCompileCheck(creation_context.GetGpuInfo(), in Compile()
201 RETURN_IF_ERROR(cl_args_.Init(creation_context.GetGpuInfo(), in CompileDeserialized()
|
| D | cl_operation.h | 41 const GpuInfo& GetGpuInfo() const { return device->info_; } in GetGpuInfo() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | convolution_transposed_3x3_thin_test.cc | 58 CreateConvolutionTransposed3x3Thin(creation_context_.GetGpuInfo(), in TEST_F()
96 CreateConvolutionTransposed3x3Thin(creation_context_.GetGpuInfo(), in TEST_F()
|
| D | convolution_transposed_thin_test.cc | 59 creation_context_.GetGpuInfo(), op_def, attr); in TEST_F()
99 creation_context_.GetGpuInfo(), op_def, attr); in TEST_F()
|
| D | conv_buffer_1x1_test.cc | 60 creation_context_.GetGpuInfo(), op_def, attr, &src_tensor.shape); in TEST_F()
97 creation_context_.GetGpuInfo(), op_def, attr, &src_tensor.shape); in TEST_F()
|
| D | conv_constants_test.cc | 59 CreateConvConstants(creation_context_.GetGpuInfo(), op_def, attr); in TEST_F()
96 CreateConvConstants(creation_context_.GetGpuInfo(), op_def, attr); in TEST_F()
|
| D | convolution_transposed_3x3_test.cc | 58 creation_context_.GetGpuInfo(), op_def, attr); in TEST_F()
|
| D | cl_test.h | 51 const GpuInfo& GetGpuInfo() const override;
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | conv_test.mm | 68 CreateConvolutionMetal(op_def, BHWC(1, 2, 2, 2), attr, env->GetGpuInfo());
104 CreateConvolutionMetal(op_def, BHWC(1, 1, 1, 1), attr, env->GetGpuInfo());
140 CreateConvolutionMetal(op_def, BHWC(1, 1, 1, 1), attr, env->GetGpuInfo());
176 CreateConvolutionMetal(op_def, BHWC(1, 2, 1, 2), attr, env->GetGpuInfo());
212 CreateConvolutionMetal(op_def, BHWC(1, 2, 2, 1), attr, env->GetGpuInfo());
270 auto gpu_op0 = CreateConvolutionMetal(op_def, dst_shape, attr, env->GetGpuInfo());
280 CreateConvolutionMetalWino4x4To6x6(op_def, conv_shape, attr, env->GetGpuInfo());
|
| D | test_util.h | 45 const GpuInfo& GetGpuInfo() const { return device_.GetInfo(); } in GetGpuInfo() function
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | testing_util.h | 43 virtual const GpuInfo& GetGpuInfo() const = 0;
|