Searched refs:kNumSplits (Results 1 – 1 of 1) sorted by relevance
247 static constexpr int64 kNumSplits = 8; member in xla::__anon6418faad0111::ConvolutionVisitor336 if (c.halo_size > CeilOfRatio(c.spatial_size, kNumSplits)) { in IsConvSuitableForSpaceToBatch()349 activations->shape().dimensions(activations_batch_dim) / kNumSplits; in HaloDuplicateWithSlice()354 new_dimensions[activations_batch_dim] = kNumSplits; in HaloDuplicateWithSlice()462 new_dimensions[activations_batch_dim] = original_batch_size * kNumSplits; in HaloDuplicateWithSlice()845 if (rhs_batch * kNumSplits != lhs_batch) { in CanPropagate()873 if (rhs_batch != kNumSplits * lhs_batch) { in CanPropagate()1031 final_shape_dims[new_space_dim] *= kNumSplits; in PropagateOnBroadcast()1908 const int64 num_splits = kNumSplits; in PropagateOnConv()2141 new_producer->shape().dimensions(new_space_dim), kNumSplits); in PropagateOnConstant()[all …]