Searched defs:spatial_dimension_to_split (Results 1 – 1 of 1) sorted by relevance
540 int64_t spatial_dimension_to_split = activations_batch_dim + 2 * (i + 1); in HaloDuplicateWithSlice() local650 int64_t spatial_dimension_to_split = spatial_dimensions_to_split->at(0); in BringSpaceNextToBatch() local2879 for (auto spatial_dimension_to_split : spatial_dimensions_to_split) { in PerformSplitSpace() local2884 for (auto spatial_dimension_to_split : spatial_dimensions_to_split) { in PerformSplitSpace() local2913 for (auto spatial_dimension_to_split : spatial_dimensions_to_split) { in PadAndSplitSpace() local3213 int64_t spatial_dimension_to_split = spatial_dimensions_to_split[0]; in PropagateOnBackpropFilterConv() local3564 const int64_t spatial_dimension_to_split = spatial_dimensions_to_split[0]; in GetConvolutionDetails() local