Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter_test.cc109 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
149 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
184 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
214 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
243 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
288 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
339 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
373 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
425 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
475 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.cc133 HloInstruction::CreateConvolve( in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.cc240 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleBatchGroupCount()
380 auto new_convolution = HloInstruction::CreateConvolve( in HandleConvolution()
442 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleConvolution()
520 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleConvolution()
Dtranspose_folding_test.cc246 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
302 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
363 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
430 HloInstruction* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Ddynamic_padder_test.cc128 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
171 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dtranspose_folding.cc177 auto new_conv = HloInstruction::CreateConvolve( in FoldTransposeIntoConvolution()
Dhlo_evaluator_test.cc927 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
982 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1066 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1144 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1204 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1268 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1340 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
1411 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
Dhlo_creation_utils.cc92 return computation->AddInstruction(HloInstruction::CreateConvolve( in MakeConvolveHlo()
Ddynamic_dimension_inference_test.cc312 auto* conv = builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dhlo_instruction.h437 static std::unique_ptr<HloInstruction> CreateConvolve(
Dalgebraic_simplifier_test.cc1371 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
3123 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
3240 builder.AddInstruction(HloInstruction::CreateConvolve( in TEST_P()
3392 b.AddInstruction(HloInstruction::CreateConvolve( in TEST_F()
Dhlo_instruction.cc421 instruction = CreateConvolve( in CreateFromProto()
780 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConvolve( in CreateConvolve() function in xla::HloInstruction
Dhlo_parser.cc1100 instruction = builder->AddInstruction(HloInstruction::CreateConvolve( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.cc545 b.AddInstruction(HloInstruction::CreateConvolve( in ConvArray4DGeneralDimensionsDilated()