Searched defs:output_tuple (Results 1 – 8 of 8) sorted by relevance
67 xla::XlaOp output_tuple = TopK(context->Input(0), k); in Compile() local
88 xla::XlaOp output_tuple = ApproxTopKFn( in Compile() local
569 xla::XlaOp output_tuple = TopK(scores_included, output_size); in ComputeResult() local
302 const RefPtr<XRTTupleAllocation>& output_tuple, in RebuildOutputAliases()350 RefPtr<XRTTupleAllocation> output_tuple, in CreateExecuteOutput()
151 XRTTupleAllocation* output_tuple; in AllocateOutputTuple() local
270 XRTTupleAllocation* output_tuple; in CreateOutputTuple() local
300 HloInstruction* output_tuple = in TEST_F() local
5662 auto output_tuple = op::Tuple( in TEST_F() local5766 auto output_tuple = op::Tuple( in TEST_F() local5881 auto output_tuple = in TEST_F() local6011 auto output_tuple = op::Tuple( in TEST_F() local6121 auto output_tuple = op::Tuple( in TEST_F() local