Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc227 new_conv->set_convolution_dimension_numbers( in TryRewriteToCudnnForwardRelu()
Dcudnn_conv_rewriter.cc61 custom_call->set_convolution_dimension_numbers(dnums); in CreateCudnnConv()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.h1041 void set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function
1196 void set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function
Dhlo_instruction.cc467 ->set_convolution_dimension_numbers( in CreateFromProto()
3449 void HloInstruction::set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function in xla::HloInstruction
3452 convolution->set_convolution_dimension_numbers(dnums); in set_convolution_dimension_numbers()
3454 custom_call->set_convolution_dimension_numbers(dnums); in set_convolution_dimension_numbers()
Dhlo_instruction_test.cc1716 instr1->set_convolution_dimension_numbers(dnums); in TEST_F()
1737 instr->set_convolution_dimension_numbers(dnums); in TEST_F()
Dhlo_instruction.h1554 void set_convolution_dimension_numbers(
Dhlo_instructions.cc2189 cloned->set_convolution_dimension_numbers(*convolution_dimension_numbers_); in CloneWithNewOperandsImpl()
Dhlo_parser.cc1551 instruction->set_convolution_dimension_numbers(*dnums); in ParseInstructionRhs()