Searched defs:spatial_dimension_to_split (Results 1 – 1 of 1) sorted by relevance
71 int64 spatial_dimension_to_split, inherent_low_padding, member344 HloInstruction* activations, int64 spatial_dimension_to_split, in HaloDuplicateWithSlice()480 int64& spatial_dimension_to_split, int64& activations_batch_dim, in BringSpaceNextToBatch()2038 HloInstruction* activations, int64 spatial_dimension_to_split, in SplitSpaceHelper()2093 int64& spatial_dimension_to_split, in SplitSpace()2324 int64 spatial_dimension_to_split = in PropagateOnBackpropFilterConv() local2601 const int64 spatial_dimension_to_split = in GetConvolutionDetails() local2716 int64 spatial_dimension_to_split = c.spatial_dimension_to_split; in PerformSpaceToBatchOnConvolution() local