Home
last modified time | relevance | path

Searched refs:transpose_params (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc177 TransposeParams transpose_params; in InitializeWeightsNodes() local
178 transpose_params.perm_count = 4; in InitializeWeightsNodes()
179 transpose_params.perm[0] = 1; in InitializeWeightsNodes()
180 transpose_params.perm[1] = 2; in InitializeWeightsNodes()
181 transpose_params.perm[2] = 3; in InitializeWeightsNodes()
182 transpose_params.perm[3] = 0; in InitializeWeightsNodes()
185 optimized_ops::Transpose<int8_t>(transpose_params, nhwc_shape, in InitializeWeightsNodes()
194 optimized_ops::Transpose<uint8_t>(transpose_params, nhwc_shape, in InitializeWeightsNodes()
Dmatmul_builder.cc175 TransposeParams transpose_params; in PopulateSubGraph() local
176 transpose_params.perm_count = 4; in PopulateSubGraph()
177 transpose_params.perm[0] = 0; in PopulateSubGraph()
178 transpose_params.perm[1] = 1; in PopulateSubGraph()
179 transpose_params.perm[2] = 3; in PopulateSubGraph()
180 transpose_params.perm[3] = 2; in PopulateSubGraph()
182 optimized_ops::Transpose<int8_t>(transpose_params, nhwc_shape, in PopulateSubGraph()
191 optimized_ops::Transpose<uint8_t>(transpose_params, nhwc_shape, in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv.cc206 TransposeParams transpose_params; in ResizeAndTransposeWeights() local
207 transpose_params.perm_count = 4; in ResizeAndTransposeWeights()
208 transpose_params.perm[0] = 1; in ResizeAndTransposeWeights()
209 transpose_params.perm[1] = 2; in ResizeAndTransposeWeights()
210 transpose_params.perm[2] = 0; in ResizeAndTransposeWeights()
211 transpose_params.perm[3] = 3; in ResizeAndTransposeWeights()
214 optimized_ops::Transpose(transpose_params, input_shape, in ResizeAndTransposeWeights()
219 optimized_ops::Transpose(transpose_params, input_shape, in ResizeAndTransposeWeights()
225 optimized_ops::Transpose(transpose_params, input_shape, in ResizeAndTransposeWeights()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data_test.cc64 TfLiteTransposeParams transpose_params; in TEST() local