Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv_3x3.cc28 bool weights_are_buffer, in DepthwiseConv3x3()
55 const OperationDef& op_def, bool weights_are_buffer, in GenerateDepthwiseConvCode()
314 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthwiseConv3x3() local
Dconvolution_transposed.cc34 const GpuInfo& gpu_info, bool weights_are_buffer) in ConvolutionTransposed()
80 bool weights_are_buffer) in ConvolutionTransposed()
129 bool weights_are_buffer, const int4& block_size) { in GenerateConvolutionTransposedCode()
581 const bool weights_are_buffer = gpu_info.IsMali() || gpu_info.IsApple(); in CreateConvolutionTransposed() local
598 const bool weights_are_buffer = gpu_info.IsMali() || gpu_info.IsApple(); in CreateConvolutionTransposed3D() local
615 const bool weights_are_buffer = gpu_info.IsMali(); in CreateConvolutionTransposedDynamicWeights() local
Ddepthwise_conv.cc71 bool weights_are_buffer, bool dynamic_weights, GPUOperation* op) { in GenerateDepthwiseConvolutionCode()
239 bool weights_are_buffer = in CreateDepthwiseConvolution2D() local
303 bool weights_are_buffer = in CreateDepthwiseConvolution3D() local
Ddepthwise_conv_3x3_stride_h2.cc28 bool weights_are_buffer, in GetKernelDepthWiseConv3x3StrideH2()
211 bool weights_are_buffer = !gpu_info.SupportsImages() || in CreateDepthWiseConv3x3StrideH2() local
Dconvolution_transposed.h99 const tflite::gpu::Tensor<OHWI, T>& weights, bool weights_are_buffer) { in UploadWeights()
142 const tflite::gpu::Tensor<OHWDI, T>& weights, bool weights_are_buffer) { in UploadWeights()
Ddepthwise_conv.h67 bool weights_are_buffer, in UploadWeightsForDWConv2D()
141 bool weights_are_buffer, in UploadWeightsForDWConv3D()
Dfully_connected.cc86 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFullyConnectedKernelCode() local
Ddepthwise_conv_3x3.h79 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases()
Ddepthwise_conv_3x3_stride_h2.h73 const tflite::gpu::Tensor<Linear, T>& biases, bool weights_are_buffer) { in UploadWeightsAndBiases()
Dfully_connected.h149 bool weights_are_buffer) { in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.cc79 const bool weights_are_buffer = UseBufferForWeights(gpu_info); in GetFCFCAddKernelCode() local
Dfc_fc_add.h126 const std::string& name, bool weights_are_buffer) { in UploadWeights()