Searched defs:output_tuple (Results 1 – 10 of 10) sorted by relevance
302 const RefPtr<XRTTupleAllocation>& output_tuple, in RebuildOutputAliases()350 RefPtr<XRTTupleAllocation> output_tuple, in CreateExecuteOutput()
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
195 PyObject* output_tuple = PyTuple_New(3); in THCPModule_getCurrentStream_wrap() local228 PyObject* output_tuple = PyTuple_New(3); in THCPModule_getDefaultStream_wrap() local408 PyObject* output_tuple = PyTuple_New(num_outputs); in THCPModule_cudaJiteratorCompileAndLaunchKernel() local
116 PyObject* output_tuple = PyTuple_New(3); in THXPModule_getCurrentStream_wrap() local
270 XRTTupleAllocation* output_tuple; in CreateOutputTuple() local
151 XRTTupleAllocation* output_tuple; in AllocateOutputTuple() 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
300 HloInstruction* output_tuple = in TEST_F() local