Searched refs:batch_increased_reshape (Results 1 – 1 of 1) sorted by relevance
2084 TF_ASSIGN_OR_RETURN(HloInstruction * batch_increased_reshape, in SplitSpaceHelper()2087 return batch_increased_reshape; in SplitSpaceHelper()2726 HloInstruction* batch_increased_reshape = retval.first; in PerformSpaceToBatchOnConvolution() local2727 convolution->SetupDerivedInstruction(batch_increased_reshape); in PerformSpaceToBatchOnConvolution()2729 VLOG(1) << "First reshape done " << batch_increased_reshape->ToString(); in PerformSpaceToBatchOnConvolution()2732 activations, HaloDuplicateWithSlice(batch_increased_reshape, in PerformSpaceToBatchOnConvolution()