Home
last modified time | relevance | path

Searched refs:new_conv (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc264 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
265 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
267 new_conv->convolution_dimension_numbers() in TEST_F()
270 new_conv->convolution_dimension_numbers() in TEST_F()
320 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
321 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
323 new_conv->convolution_dimension_numbers() in TEST_F()
326 new_conv->convolution_dimension_numbers() in TEST_F()
330 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0)); in TEST_F()
333 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1)); in TEST_F()
[all …]
Dtranspose_folding.cc177 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution() local
182 &convolution, std::move(new_conv))); in FoldTransposeIntoConvolution()
Dconvolution_group_converter.cc529 auto new_conv = HloInstruction::CreateConcatenate( in HandleConvolution() local
532 convolution, std::move(new_conv))); in HandleConvolution()
Dalgebraic_simplifier.cc3452 auto new_conv = convolution->CloneWithNewOperands( in FoldConvInputPad() local
3454 new_conv->set_window(new_window); in FoldConvInputPad()
3456 ReplaceWithNewInstruction(convolution, std::move(new_conv))); in FoldConvInputPad()
3518 auto new_conv = convolution->CloneWithNewOperands( in FoldConvFilterPad() local
3520 new_conv->set_window(new_window); in FoldConvFilterPad()
3522 ReplaceWithNewInstruction(convolution, std::move(new_conv))); in FoldConvFilterPad()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc224 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu() local
226 new_conv->set_window(conv->window()); in TryRewriteToCudnnForwardRelu()
227 new_conv->set_convolution_dimension_numbers( in TryRewriteToCudnnForwardRelu()
229 new_conv->set_metadata(conv->metadata()); in TryRewriteToCudnnForwardRelu()
236 TF_RETURN_IF_ERROR(new_conv->set_backend_config(config)); in TryRewriteToCudnnForwardRelu()
239 << new_conv->ToString(); in TryRewriteToCudnnForwardRelu()
241 new_conv, 0); in TryRewriteToCudnnForwardRelu()
Dcudnn_conv_pad_for_tensor_cores.cc95 auto* new_conv = in PadConv() local
107 HloInstruction::CreateGetTupleElement(new_result_shape, new_conv, 0)); in PadConv()
112 new_conv = in PadConv()
117 << new_conv->ToString(); in PadConv()
118 return conv->parent()->ReplaceInstruction(conv, new_conv); in PadConv()
Dcudnn_conv_padding_legalization.cc183 auto new_conv = conv->parent()->AddInstruction( in CanonicalizeForwardConvolution() local
185 new_conv->set_window(new_conv_window); in CanonicalizeForwardConvolution()
187 << new_conv->ToString(); in CanonicalizeForwardConvolution()
188 TF_CHECK_OK(conv->parent()->ReplaceInstruction(conv, new_conv)); in CanonicalizeForwardConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc132 HloInstruction* new_conv = module->entry_computation()->AddInstruction( in Run() local
141 hlo->shape(), new_conv, in Run()