Home
last modified time | relevance | path

Searched defs:output_tuple (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtopk_op.cc51 xla::XlaOp output_tuple = TopK(context->Input(0), k); in Compile() local
Dimage_ops.cc545 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile() local
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc299 const RefPtr<XRTTupleAllocation>& output_tuple, in RebuildOutputAliases()
347 RefPtr<XRTTupleAllocation> output_tuple, in CreateExecuteOutput()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc151 XRTTupleAllocation* output_tuple; in AllocateOutputTuple() local
Dxrt_execute_op.cc270 XRTTupleAllocation* output_tuple; in CreateOutputTuple() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc299 HloInstruction* output_tuple = in TEST_F() local