Searched refs:transposed_weights (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_conv.cc | 192 TfLiteTensor* transposed_weights) { in ResizeAndTransposeWeights() argument
200 transposed_weights->type = weights->type; in ResizeAndTransposeWeights()
201 transposed_weights->allocation_type = kTfLiteDynamic; in ResizeAndTransposeWeights()
202 TF_LITE_ENSURE_STATUS(context->ResizeTensor(context, transposed_weights, in ResizeAndTransposeWeights()
216 GetTensorShape(transposed_weights), in ResizeAndTransposeWeights()
217 GetTensorData<float>(transposed_weights)); in ResizeAndTransposeWeights()
221 GetTensorShape(transposed_weights), in ResizeAndTransposeWeights()
222 GetTensorData<uint8>(transposed_weights)); in ResizeAndTransposeWeights()
227 GetTensorShape(transposed_weights), in ResizeAndTransposeWeights()
228 GetTensorData<int8>(transposed_weights)); in ResizeAndTransposeWeights()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | matmul_builder.cc | 251 auto transposed_weights = AddOutput(sizeof(uint8_t), 4, weights_shape_); in PopulateSubGraph() local
258 AddFullyConnected(inputs, outputs, transposed_weights, transposed_weights_min, in PopulateSubGraph()
|