Searched refs:RearrangeFCWeightsToIOO4I4 (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | fully_connected_test.cc | 82 RearrangeFCWeightsToIOO4I4(weights, data.data()); in TEST_F()
123 RearrangeFCWeightsToIOO4I4(weights, data.data()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | fully_connected.h | 40 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4() function
167 RearrangeFCWeightsToIOO4I4(weights, ptr); in UploadWeights()
170 RearrangeFCWeightsToIOO4I4(weights, ptr); in UploadWeights()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | fc_fc_add.h | 40 void RearrangeFCWeightsToIOO4I4(const tflite::gpu::Tensor<OHWI, T>& weights, in RearrangeFCWeightsToIOO4I4() function
144 RearrangeFCWeightsToIOO4I4(weights, ptr); in UploadWeights()
147 RearrangeFCWeightsToIOO4I4(weights, ptr); in UploadWeights()
|