Home
last modified time | relevance | path

Searched defs:output_tuple (Results 1 – 5 of 5) 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.cc539 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc142 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormForwardTrainingThunk()
222 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormBackwardThunk()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc227 XRTTupleAllocation* output_tuple; in DoWork() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc298 HloInstruction* output_tuple = in TEST_F() local