Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc266 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
267 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
269 new_conv->convolution_dimension_numbers() in TEST_F()
272 new_conv->convolution_dimension_numbers() in TEST_F()
323 HloInstruction* new_conv = *instruction_set.begin(); in TEST_F() local
324 EXPECT_EQ(HloOpcode::kConvolution, new_conv->opcode()); in TEST_F()
326 new_conv->convolution_dimension_numbers() in TEST_F()
329 new_conv->convolution_dimension_numbers() in TEST_F()
333 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(0)); in TEST_F()
336 new_conv->convolution_dimension_numbers().kernel_spatial_dimensions(1)); in TEST_F()
[all …]
Dspace_to_batch_converter.cc2015 HloInstruction * new_conv, in PropagateOnConv()
2021 convolution->SetupDerivedInstruction(new_conv); in PropagateOnConv()
2023 old_to_new_instrs_[convolution] = new_conv; in PropagateOnConv()
2024 VLOG(1) << "Space-to-batched convolution " << new_conv->ToString(); in PropagateOnConv()
2031 instr_to_dim_permute_map_[new_conv] = std::vector<int64>(transpose_dims); in PropagateOnConv()
2530 HloInstruction * new_conv, in PropagateOnBackpropFilterConv()
2536 convolution->SetupDerivedInstruction(new_conv); in PropagateOnBackpropFilterConv()
2538 VLOG(2) << "New backprop filter convolution " << new_conv->ToString(); in PropagateOnBackpropFilterConv()
2540 std::vector<int64> output_sizes(new_conv->shape().dimensions().begin(), in PropagateOnBackpropFilterConv()
2541 new_conv->shape().dimensions().end()); in PropagateOnBackpropFilterConv()
[all …]
Dtranspose_folding.cc206 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution() local
211 &convolution, std::move(new_conv))); in FoldTransposeIntoConvolution()
Dalgebraic_simplifier.cc5201 auto new_conv = convolution->CloneWithNewOperands( in FoldConvInputPad() local
5203 new_conv->set_window(new_window); in FoldConvInputPad()
5205 ReplaceWithNewInstruction(convolution, std::move(new_conv))); in FoldConvInputPad()
5267 auto new_conv = convolution->CloneWithNewOperands( in FoldConvFilterPad() local
5269 new_conv->set_window(new_window); in FoldConvFilterPad()
5271 ReplaceWithNewInstruction(convolution, std::move(new_conv))); in FoldConvFilterPad()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc238 auto new_conv = computation->AddInstruction(HloInstruction::CreateCustomCall( in TryRewriteToCudnnForwardRelu() local
240 new_conv->set_feature_group_count(conv->feature_group_count()); in TryRewriteToCudnnForwardRelu()
241 new_conv->set_window(conv->window()); in TryRewriteToCudnnForwardRelu()
242 new_conv->set_convolution_dimension_numbers( in TryRewriteToCudnnForwardRelu()
244 new_conv->set_metadata(conv->metadata()); in TryRewriteToCudnnForwardRelu()
251 TF_RETURN_IF_ERROR(new_conv->set_backend_config(config)); in TryRewriteToCudnnForwardRelu()
254 << new_conv->ToString(); in TryRewriteToCudnnForwardRelu()
256 new_conv, 0); in TryRewriteToCudnnForwardRelu()
Dcudnn_pad_for_convolutions.cc100 auto* new_conv = in PadConv() local
112 HloInstruction::CreateGetTupleElement(new_result_shape, new_conv, 0)); in PadConv()
117 new_conv = in PadConv()
122 << new_conv->ToString(); in PadConv()
123 return conv->parent()->ReplaceInstruction(conv, new_conv); in PadConv()
Dgpu_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()
Dgpu_conv_rewriter.cc114 auto new_conv = conv->CloneWithNewOperands(output_shape, new_operands); in ConvertBatchGroupedToFeatureGroupedConvolution() local
115 new_conv->set_feature_group_count(num_groups); in ConvertBatchGroupedToFeatureGroupedConvolution()
116 new_conv->set_batch_group_count(1); in ConvertBatchGroupedToFeatureGroupedConvolution()
117 new_conv->set_convolution_dimension_numbers(dim_numbers); in ConvertBatchGroupedToFeatureGroupedConvolution()
118 return computation->AddInstruction(std::move(new_conv)); in ConvertBatchGroupedToFeatureGroupedConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc133 HloInstruction* new_conv = module->entry_computation()->AddInstruction( in Run() local
142 hlo->shape(), new_conv, in Run()