Searched refs:BringSpaceNextToBatch (Results 1 – 1 of 1) sorted by relevance
161 StatusOr<SpaceNextToBatchDetails> BringSpaceNextToBatch(478 ConvolutionVisitor::BringSpaceNextToBatch( in BringSpaceNextToBatch() function in xla::__anon6418faad0111::ConvolutionVisitor1889 BringSpaceNextToBatch( in PropagateOnConv()2099 BringSpaceNextToBatch( in SplitSpace()2348 BringSpaceNextToBatch(activations_new, permuted_conv_dims_numbers, in PropagateOnBackpropFilterConv()