HomeSort by relevance Sort by last modified time
    Searched defs:tuple_param (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
tuple_simplifier_test.cc 174 HloInstruction* tuple_param = builder.AddInstruction( local
177 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0));
179 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1));
181 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 2));
192 EXPECT_THAT(computation->root_instruction(), tuple_param);
199 HloInstruction* tuple_param = builder.AddInstruction( local
202 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 0));
204 HloInstruction::CreateGetTupleElement(scalar_shape_, tuple_param, 1));
262 HloInstruction* tuple_param = builder.AddInstruction( local
265 HloInstruction::CreateCall(tuple_shape_, {tuple_param}, c0))
    [all...]
hlo_dataflow_analysis_test.cc 469 // body((F32[], F32[]) %tuple_param):
470 // %add = Add(%tuple_param{0}, %tuple_param{1})
471 // return Tuple(%tuple_param{0}, %add)
473 // condition((F32[], F32[]) %tuple_param):
570 // body((F32[], F32[]) %tuple_param):
571 // %add = Add(%tuple_param{0}, %tuple_param{1})
572 // return Tuple(%tuple_param{0}, %add)
574 // condition((F32[], F32[]) %tuple_param)
2029 auto tuple_param = builder.AddInstruction(HloInstruction::CreateParameter( local
    [all...]

Completed in 3752 milliseconds