Home
last modified time | relevance | path

Searched refs:CreateConvolve (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_rewriter_test.cc110 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
151 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
187 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
217 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
246 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
291 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
343 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
379 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
432 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
483 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc84 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
147 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dconv_canonicalization.cc136 HloInstruction::CreateConvolve( in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtranspose_folding_test.cc254 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
312 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
375 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
444 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dconvolution_group_converter.cc340 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleBatchGroupCount()
486 auto new_convolution = HloInstruction::CreateConvolve( in HandleConvolution()
573 computation_->AddInstruction(HloInstruction::CreateConvolve( in HandleConvolution()
Dtranspose_folding.cc173 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution()
Ddot_as_convolution_util.cc175 return HloInstruction::CreateConvolve( in CreateShardedConvForDotGeneralConvolution()
Dhlo_evaluator_test.cc1042 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1097 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1181 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1259 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1319 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1383 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1455 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1526 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
Ddynamic_padder_test.cc356 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
399 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Ddynamic_padder.cc1017 custom_call_conv->AddInstruction(HloInstruction::CreateConvolve( in RewriteDynamicConvolutionInputGrad()
1078 custom_call_conv->AddInstruction(HloInstruction::CreateConvolve( in RewriteDynamicConvolutionForward()
1162 custom_call_conv->AddInstruction(HloInstruction::CreateConvolve( in RewriteDynamicConvolutionKernelGrad()
Dhlo_creation_utils.cc132 HloInstruction::CreateConvolve( in MakeConvolveHlo()
Ddynamic_dimension_inference_test.cc469 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dhlo_instruction.h616 static std::unique_ptr<HloInstruction> CreateConvolve(
Dalgebraic_simplifier_test.cc1805 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
4319 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
4438 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
4597 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dhlo_instruction.cc684 instruction = CreateConvolve( in CreateFromProto()
1170 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConvolve( in CreateConvolve() function in xla::HloInstruction
Dhlo_parser.cc2027 return builder->AddInstruction(HloInstruction::CreateConvolve( in CreateInstruction()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc483 b.AddInstruction(HloInstruction::CreateConvolve( in ConvArray4DGeneralDimensionsDilated()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc958 return HloInstruction::CreateConvolve( in CreateShardedConvConvolution()
Ddot_handler.cc1327 dot = body_b.AddInstruction(HloInstruction::CreateConvolve( in PartitionBaseCase()