Searched refs:nhcw_shape (Results 1 – 1 of 1) sorted by relevance
173 RuntimeShape nhcw_shape({batch_size, height_size, depth_size, width_size}); in PopulateSubGraph() local183 weights_tensor.data.int8, nhcw_shape, in PopulateSubGraph()192 weights_tensor.data.uint8, nhcw_shape, in PopulateSubGraph()