Searched refs:new_space_dim (Results 1 – 1 of 1) sorted by relevance
1211 const int64_t new_space_dim = in CanPropagate() local1216 if (first_operand->shape().dimensions(new_space_dim) % rhs_dilation != in CanPropagate()1375 const int64_t new_space_dim = in CanPropagate() local1392 if ((first_operand->shape().dimensions(new_space_dim) + pad_high + in CanPropagate()1395 second_operand->shape().dimensions(new_space_dim)) { in CanPropagate()1422 const int64_t new_space_dim = DimLookUp(permute_dims, old_space_dim); in PropagateOnBroadcast() local1441 final_shape_dims[new_space_dim] *= ctrl_.number_of_splits; in PropagateOnBroadcast()1696 const int64_t new_space_dim = DimLookUp(permute_dims, old_space_dim); in SupportedOpForPropagation() local1698 if (new_operand->shape().dimensions(new_space_dim) % in SupportedOpForPropagation()1967 int64_t new_batch_dim, new_space_dim, new_feature_dim; in Propagate() local[all …]