Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_rewriter.cc582 custom_call->set_convolution_dimension_numbers(dnums); in CreateGpuConv()
653 new_conv->set_convolution_dimension_numbers(dim_numbers); in ConvertBatchGroupedToFeatureGroupedConvolution()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconvolution_4d_expander.cc169 instruction->set_convolution_dimension_numbers(new_dim_nums); in ExpandInstruction()
H A Dhlo_instructions.h1605 void set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function
1838 void set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function
H A Dhlo_instruction.cc761 custom_call_instr->set_convolution_dimension_numbers( in CreateFromProto()
4728 void HloInstruction::set_convolution_dimension_numbers( in set_convolution_dimension_numbers() function in xla::HloInstruction
4731 convolution->set_convolution_dimension_numbers(dnums); in set_convolution_dimension_numbers()
4733 custom_call->set_convolution_dimension_numbers(dnums); in set_convolution_dimension_numbers()
H A Dhlo_instruction_test.cc2021 instr1->set_convolution_dimension_numbers(dnums); in TEST_F()
2054 instr->set_convolution_dimension_numbers(dnums); in TEST_F()
H A Dhlo_instruction.h2056 void set_convolution_dimension_numbers(
H A Dhlo_instructions.cc2917 cloned->set_convolution_dimension_numbers(*convolution_dimension_numbers_); in CloneWithNewOperandsImpl()
H A Dhlo_parser.cc2705 custom_call_instr->set_convolution_dimension_numbers(*dnums); in CreateInstruction()