Searched refs:get_chosen_spatial_dim (Results 1 – 1 of 1) sorted by relevance
193 int64 get_chosen_spatial_dim(HloInstruction* convolution) { in get_chosen_spatial_dim() function in xla::__anon6418faad0111::ConvolutionVisitor293 .dimensions(get_chosen_spatial_dim(convolution)) in IsConvSuitableForSpaceToBatch()301 .dimensions(get_chosen_spatial_dim(convolution)) in IsConvSuitableForSpaceToBatch()767 get_chosen_spatial_dim(consumer)) != dim_map_val_op_0.second) { in CanPropagate()786 .dimensions(get_chosen_spatial_dim(consumer)) in CanPropagate()804 consumer->window().dimensions(get_chosen_spatial_dim(consumer)); in CanPropagate()1855 get_chosen_spatial_dim(convolution)); in PropagateOnConv()1911 get_chosen_spatial_dim(convolution))); in PropagateOnConv()1990 if (j == get_chosen_spatial_dim(convolution)) { in PropagateOnConv()2010 new_window.mutable_dimensions(get_chosen_spatial_dim(convolution)) in PropagateOnConv()[all …]