Searched refs:new_space_dim (Results 1 – 1 of 1) sorted by relevance
152 HloInstruction* select_val, int64 new_batch_dim, int64 new_space_dim,833 const int64 new_space_dim = in CanPropagate() local837 if (first_operand->shape().dimensions(new_space_dim) % rhs_dilation != in CanPropagate()968 const int64 new_space_dim = in CanPropagate() local984 if ((first_operand->shape().dimensions(new_space_dim) + pad_high + in CanPropagate()987 second_operand->shape().dimensions(new_space_dim)) { in CanPropagate()1012 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBroadcast() local1031 final_shape_dims[new_space_dim] *= kNumSplits; in PropagateOnBroadcast()1184 const int64 new_space_dim = DimLookUp(permute_dims, old_space_dim); in SupportedOpForPropagation() local1188 if (new_operand->shape().dimensions(new_space_dim) % in SupportedOpForPropagation()[all …]