Searched refs:dst_weight_desc (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.cc | 41 const WeightsDescription& dst_weight_desc, DataType dst_type, in RearrangeWeights() argument
44 GetTotalElementsCountForLayout(dst_weight_desc, weights.shape); in RearrangeWeights()
45 if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupI4O4) { in RearrangeWeights()
49 dst_weight_desc.output_group_size, in RearrangeWeights()
54 dst_weight_desc.output_group_size, in RearrangeWeights()
58 } else if (dst_weight_desc.layout == WeightsLayout::kOHWIOGroupO4I4) { in RearrangeWeights()
62 dst_weight_desc.output_group_size, in RearrangeWeights()
67 dst_weight_desc.output_group_size, in RearrangeWeights()
71 } else if (dst_weight_desc.layout == WeightsLayout::kOICustomSpatialI4O4) { in RearrangeWeights()
75 weights, dst_weight_desc.spatial_remap, in RearrangeWeights()
[all …]
|
| D | weights_conversion.h | 332 const WeightsDescription& dst_weight_desc, DataType dst_type,
|