Searched defs:nhwc_shape (Results 1 – 2 of 2) sorted by relevance
172 RuntimeShape nhwc_shape({weights_batch_size, weights_height_size, in InitializeWeightsNodes() local
172 RuntimeShape nhwc_shape({batch_size, height_size, width_size, depth_size}); in PopulateSubGraph() local