Home
last modified time | relevance | path

Searched refs:RearrangeFCWeightsToOIO4I4 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dfully_connected.h91 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4() function
183 RearrangeFCWeightsToOIO4I4(weights, ptr); in UploadWeights()
186 RearrangeFCWeightsToOIO4I4(weights, ptr); in UploadWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h67 void RearrangeFCWeightsToOIO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToOIO4I4() function
163 RearrangeFCWeightsToOIO4I4(weights, ptr); in UploadWeights()
166 RearrangeFCWeightsToOIO4I4(weights, ptr); in UploadWeights()