Searched defs:new_conv (Results 1 – 9 of 9) sorted by relevance
139 HloInstruction* new_conv = comp->AddInstruction( in EnsureIsConvBiasActivation() local173 HloInstruction* new_conv = in FuseConvertToFloat() local321 HloInstruction* new_conv = comp->AddInstruction( in FuseBiasOrSideInput() local434 HloInstruction* new_conv = comp->AddInstruction( in FuseSideInputAlpha() local532 HloInstruction* new_conv = comp->AddInstruction( in FuseConvertToF16() local613 HloInstruction* new_conv = comp->AddInstruction( in FuseConvertToS8() local
345 XlaOp new_conv = CustomCallWithConvDnums( in TryRevectorizeConv() local466 XlaOp new_conv = CustomCallWithConvDnums( in TryVectorizeConv() local
186 auto new_conv = conv->parent()->AddInstruction( in CanonicalizeForwardConvolution() local
97 auto* new_conv = in PadConv() local
650 auto new_conv = conv->CloneWithNewOperands(output_shape, new_operands); in ConvertBatchGroupedToFeatureGroupedConvolution() local
273 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local331 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local394 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local463 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
173 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution() local
6329 auto new_conv = in FoldConvInputPad() local6397 auto new_conv = convolution->CloneWithNewOperands( in FoldConvFilterPad() local
135 HloInstruction* new_conv = module->entry_computation()->AddInstruction( in Run() local