Home
last modified time | relevance | path

Searched refs:wino_weights (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_buffer_1x1.h116 tflite::gpu::Tensor<OHWI, T> wino_weights; in UploadDataForWinograd4x4To6x6() local
117 RearrangeWeightsToWinograd4x4To6x6Weights(weights, &wino_weights); in UploadDataForWinograd4x4To6x6()
118 UploadWeights(wino_weights); in UploadDataForWinograd4x4To6x6()
Dconv_powervr.h215 tflite::gpu::Tensor<OHWI, T> wino_weights; in UploadDataForWinograd4x4To6x6() local
216 RearrangeWeightsToWinograd4x4To6x6Weights(weights, &wino_weights); in UploadDataForWinograd4x4To6x6()
217 UploadWeights(wino_weights); in UploadDataForWinograd4x4To6x6()
Dconv_metal.cc1149 tflite::gpu::Tensor<OHWI, DataType::FLOAT32> wino_weights; in CreateConvolutionMetalWino4x4To6x6() local
1151 RearrangeWeightsToWinograd4x4To6x6Weights(attr.weights, &wino_weights); in CreateConvolutionMetalWino4x4To6x6()
1159 wino_weights, desc.GetWeightsDescription(), weights_type); in CreateConvolutionMetalWino4x4To6x6()