Searched refs:nhwc_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 172 RuntimeShape nhwc_shape({weights_batch_size, weights_height_size, in InitializeWeightsNodes() local
185 optimized_ops::Transpose<int8_t>(transpose_params, nhwc_shape, in InitializeWeightsNodes()
194 optimized_ops::Transpose<uint8_t>(transpose_params, nhwc_shape, in InitializeWeightsNodes()
|
| D | matmul_builder.cc | 172 RuntimeShape nhwc_shape({batch_size, height_size, width_size, depth_size}); in PopulateSubGraph() local
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/python/framework/ |
| D | test_util.py | 412 nhwc_shape = [input_shape[a] for a in permutation[:-1]]
413 nhwc_shape[-1] *= input_shape[-1]
416 return array_ops.reshape(t, nhwc_shape)
418 return nhwc_shape
|