| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | conv_buffer_1x1.h | 94 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights);
109 UploadWeights(weights); in UploadData()
118 UploadWeights(wino_weights); in UploadDataForWinograd4x4To6x6()
126 void ConvBuffer1x1::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights() function
|
| D | convolution_transposed.h | 81 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights,
85 void UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights,
98 void ConvolutionTransposed::UploadWeights( in UploadWeights() function
141 void ConvolutionTransposed::UploadWeights( in UploadWeights() function
|
| D | conv_powervr.h | 134 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights);
137 void UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights);
208 UploadWeights(weights); in UploadData()
217 UploadWeights(wino_weights); in UploadDataForWinograd4x4To6x6()
255 void ConvPowerVR::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights) { in UploadWeights() function
299 void ConvPowerVR::UploadWeights(const tflite::gpu::Tensor<OHWDI, T>& weights) { in UploadWeights() function
|
| D | fully_connected.h | 140 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights,
148 void FullyConnected::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights() function
|
| D | convolution_transposed_3x3_thin.h | 71 void UploadWeights(
|
| D | convolution_transposed_3x3_thin.cc | 203 void ConvolutionTransposed3x3Thin::UploadWeights( in UploadWeights() function in tflite::gpu::ConvolutionTransposed3x3Thin
239 result.UploadWeights(attr.weights); in CreateConvolutionTransposed3x3Thin()
|
| D | convolution_transposed_3x3.h | 78 void UploadWeights(
|
| D | convolution_transposed_4x4.h | 81 void UploadWeights(
|
| D | fully_connected.cc | 161 result.UploadWeights(attr.weights, UseBufferForWeights(gpu_info)); in CreateFullyConnected()
|
| D | convolution_transposed_4x4.cc | 391 void ConvolutionTransposed4x4::UploadWeights( in UploadWeights() function in tflite::gpu::ConvolutionTransposed4x4
430 result.UploadWeights(attr.weights, GetBestWeightsUploadType(gpu_info)); in CreateConvolutionTransposed4x4()
|
| D | convolution_transposed_3x3.cc | 387 void ConvolutionTransposed3x3::UploadWeights( in UploadWeights() function in tflite::gpu::ConvolutionTransposed3x3
426 result.UploadWeights(attr.weights); in CreateConvolutionTransposed3x3()
|
| D | convolution_transposed.cc | 583 result.UploadWeights(attr.weights, weights_are_buffer); in CreateConvolutionTransposed()
600 result.UploadWeights(attr.weights, weights_are_buffer); in CreateConvolutionTransposed3D()
|
| D | conv_powervr.cc | 1482 result.UploadWeights(attr.weights); in CreateConvPowerVR3D()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.cc | 177 result.UploadWeights(attr0.weights, "weights0", in CreateFCFCAdd()
179 result.UploadWeights(attr1.weights, "weights1", in CreateFCFCAdd()
|
| D | fc_fc_add.h | 117 void UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights,
125 void FCFCAdd::UploadWeights(const tflite::gpu::Tensor<OHWI, T>& weights, in UploadWeights() function
|
| D | depthwise_conv_plus_1x1_conv.cc | 27 void UploadWeights(const DepthwiseConvolution2DAttributes& dw_attr, in UploadWeights() function
262 UploadWeights(dw_attr, conv_attr, definition.precision, &result); in CreateDepthwiseConvPlus1x1Conv()
|