| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv_3x3_stride_h2.cc | 28 bool weights_are_buffer, in GetKernelDepthWiseConv3x3StrideH2() argument
58 } else if (weights_are_buffer) { in GetKernelDepthWiseConv3x3StrideH2()
76 if (!weights_are_buffer) { in GetKernelDepthWiseConv3x3StrideH2()
109 if (local_mem_uploads || weights_are_buffer) { in GetKernelDepthWiseConv3x3StrideH2()
168 if (!weights_are_buffer) { in GetKernelDepthWiseConv3x3StrideH2()
211 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthWiseConv3x3StrideH2() local
216 desc.local_mem_uploads_ = weights_are_buffer && gpu_info.IsPowerVR(); in CreateDepthWiseConv3x3StrideH2()
218 desc.code_ = GetKernelDepthWiseConv3x3StrideH2(definition, weights_are_buffer, in CreateDepthWiseConv3x3StrideH2()
230 desc.UploadWeightsAndBiases(attr.weights, attr.bias, weights_are_buffer); in CreateDepthWiseConv3x3StrideH2()
|
| D | convolution_transposed.h | 75 const GpuInfo& gpu_info, bool weights_are_buffer);
78 const GpuInfo& gpu_info, bool weights_are_buffer);
82 bool weights_are_buffer);
86 bool weights_are_buffer);
90 bool weights_are_buffer,
99 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeights() argument
110 if (weights_are_buffer) { in UploadWeights()
142 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeights() argument
159 if (weights_are_buffer) { in UploadWeights()
168 if (weights_are_buffer) { in UploadWeights()
[all …]
|
| D | depthwise_conv_3x3.cc | 28 bool weights_are_buffer, in DepthwiseConv3x3() argument
33 code_ = GenerateDepthwiseConvCode(definition_, weights_are_buffer, in DepthwiseConv3x3()
55 const OperationDef& op_def, bool weights_are_buffer, in GenerateDepthwiseConvCode() argument
98 } else if (weights_are_buffer) { in GenerateDepthwiseConvCode()
109 if (!weights_are_buffer) { in GenerateDepthwiseConvCode()
158 if (local_mem_uploads || weights_are_buffer) { in GenerateDepthwiseConvCode()
246 if (!weights_are_buffer) { in GenerateDepthwiseConvCode()
314 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthwiseConv3x3() local
317 bool local_mem_uploads = weights_are_buffer && gpu_info.IsPowerVR(); in CreateDepthwiseConv3x3()
318 DepthwiseConv3x3 result(definition, weights_are_buffer, local_mem_uploads, in CreateDepthwiseConv3x3()
[all …]
|
| D | convolution_transposed.cc | 34 const GpuInfo& gpu_info, bool weights_are_buffer) in ConvolutionTransposed() argument
38 if (weights_are_buffer) { in ConvolutionTransposed()
74 weights_are_buffer, block_size_); in ConvolutionTransposed()
80 bool weights_are_buffer) in ConvolutionTransposed() argument
84 if (weights_are_buffer) { in ConvolutionTransposed()
124 weights_are_buffer, block_size_); in ConvolutionTransposed()
129 bool weights_are_buffer, const int4& block_size) { in GenerateConvolutionTransposedCode() argument
159 const std::string f0 = weights_are_buffer ? "FLT16_0123(weights_cache[" + in GenerateConvolutionTransposedCode()
162 const std::string f1 = weights_are_buffer ? "FLT16_4567(weights_cache[" + in GenerateConvolutionTransposedCode()
165 const std::string f2 = weights_are_buffer ? "FLT16_89ab(weights_cache[" + in GenerateConvolutionTransposedCode()
[all …]
|
| D | depthwise_conv.cc | 71 bool weights_are_buffer, bool dynamic_weights, GPUOperation* op) { in GenerateDepthwiseConvolutionCode() argument
126 if (weights_are_buffer) { in GenerateDepthwiseConvolutionCode()
202 if (weights_are_buffer) { in GenerateDepthwiseConvolutionCode()
239 bool weights_are_buffer = in CreateDepthwiseConvolution2D() local
257 weights_are_buffer, false, &op); in CreateDepthwiseConvolution2D()
258 UploadWeightsForDWConv2D(attr.weights, weights_are_buffer, in CreateDepthwiseConvolution2D()
263 desc.storage_type = weights_are_buffer ? LinearStorageType::BUFFER in CreateDepthwiseConvolution2D()
303 bool weights_are_buffer = in CreateDepthwiseConvolution3D() local
325 weights_are_buffer, false, &op); in CreateDepthwiseConvolution3D()
326 UploadWeightsForDWConv3D(attr.weights, weights_are_buffer, in CreateDepthwiseConvolution3D()
[all …]
|
| D | depthwise_conv_3x3.h | 53 bool weights_are_buffer, bool local_mem_uploads,
58 bool weights_are_buffer);
70 bool weights_are_buffer,
79 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases() argument
98 if (weights_are_buffer) { in UploadWeightsAndBiases()
|
| D | depthwise_conv.h | 67 bool weights_are_buffer, in UploadWeightsForDWConv2D() argument
90 if (weights_are_buffer) { in UploadWeightsForDWConv2D()
141 bool weights_are_buffer, in UploadWeightsForDWConv3D() argument
165 if (weights_are_buffer) { in UploadWeightsForDWConv3D()
|
| D | fully_connected.h | 141 bool weights_are_buffer);
149 bool weights_are_buffer) { in UploadWeights() argument
158 if (weights_are_buffer) { in UploadWeights()
|
| D | depthwise_conv_3x3_stride_h2.h | 61 bool weights_are_buffer);
73 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases() argument
92 if (weights_are_buffer) { in UploadWeightsAndBiases()
|
| D | fully_connected.cc | 86 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFullyConnectedKernelCode() local
110 if (weights_are_buffer) { in GetFullyConnectedKernelCode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 79 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFCFCAddKernelCode() local
105 if (weights_are_buffer) { in GetFCFCAddKernelCode()
129 if (weights_are_buffer) { in GetFCFCAddKernelCode()
|
| D | fc_fc_add.h | 118 const std::string& name, bool weights_are_buffer);
126 const std::string& name, bool weights_are_buffer) { in UploadWeights() argument
135 if (weights_are_buffer) { in UploadWeights()
|