Searched refs:old_batch_dim (Results 1 – 1 of 1) sorted by relevance
194 absl::Span<const int64_t> new_space_dims, int64_t old_batch_dim,489 const int64_t old_batch_dim = in IsThisBackPropFilterConv() local492 .kernel_input_feature_dimension() != old_batch_dim) { in IsThisBackPropFilterConv()499 const int64_t old_batch_dim = in IsThisBackPropFilterConv() local501 if (dim_numbers.input_feature_dimension() != old_batch_dim) { in IsThisBackPropFilterConv()1202 const int64_t old_batch_dim = in CanPropagate() local1210 DimLookUp(permute_dims_first_operand, old_batch_dim); in CanPropagate()1245 const int64_t old_batch_dim = in CanPropagate() local1251 DimLookUp(permute_dims_second_operand, old_batch_dim); in CanPropagate()1368 const int64_t old_batch_dim = in CanPropagate() local[all …]