Searched defs:new_conv (Results 1 – 8 of 8) sorted by relevance
266 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local323 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local385 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local453 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
206 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution() local
5201 auto new_conv = convolution->CloneWithNewOperands( in FoldConvInputPad() local5267 auto new_conv = convolution->CloneWithNewOperands( in FoldConvFilterPad() local
133 HloInstruction* new_conv = module->entry_computation()->AddInstruction( in Run() local
100 auto* new_conv = in PadConv() local
183 auto new_conv = conv->parent()->AddInstruction( in CanonicalizeForwardConvolution() local
238 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu() local
114 auto new_conv = conv->CloneWithNewOperands(output_shape, new_operands); in ConvertBatchGroupedToFeatureGroupedConvolution() local