Home
last modified time | relevance | path

Searched defs:weights_are_buffer (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Ddepthwise_conv.cc182 const bool weights_are_buffer = UseBuffersForWeights(gpu_info); in GenerateWeightsUpload() local
184 int work_group_total_size) { in GenerateWeightsUpload()
219 const bool weights_are_buffer = UseBuffersForWeights(gpu_info); in GenerateCode() local
390 const bool weights_are_buffer = UseBuffersForWeights(gpu_info); in CreateDepthwiseConvolution2D() local
456 const bool weights_are_buffer = UseBuffersForWeights(gpu_info); in CreateDepthwiseConvolution3D() local
H A Ddepthwise_conv_3x3.cc29 bool weights_are_buffer, in DepthwiseConv3x3()
57 bool weights_are_buffer, bool local_mem_uploads) { in GenerateDepthwiseConvCode()
337 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthwiseConv3x3() local
H A Dconvolution_transposed.h101 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeights()
132 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeights()
H A Ddepthwise_conv_3x3_stride_h2.cc30 bool weights_are_buffer, in GetKernelDepthWiseConv3x3StrideH2()
217 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthWiseConv3x3StrideH2() local
H A Ddepthwise_conv.h147 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeightsForDWConv2D()
218 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeightsForDWConv3D()
H A Ddepthwise_conv_3x3.h82 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases()
H A Ddepthwise_conv_3x3_stride_h2.h75 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases()
H A Dfully_connected.h158 bool weights_are_buffer) { in UploadWeights()
H A Dfully_connected.cc106 bool weights_are_buffer, bool quantized) { in GetFullyConnectedKernelCode()
H A Dconvolution_transposed.cc158 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GenerateConvolutionTransposedCode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Dfc_fc_add.cc104 bool weights_are_buffer, in GetFCFCAddKernelCode()
256 bool weights_are_buffer = UseBufferForWeights(gpu_info); in CreateFCFCAdd() local
H A Dfc_fc_add.h140 const std::string& name, bool weights_are_buffer) { in UploadWeights()