Searched refs:output_tuple (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 227 XRTTupleAllocation* output_tuple; in DoWork() local
230 &output_tuple)); in DoWork()
243 ? output_tuple->AliasBufferFrom( in DoWork()
251 output_tuple->on_device_shape().IsTuple()) { in DoWork()
253 xla::ShapeUtil::TupleElementCount(output_tuple->on_device_shape()); in DoWork()
264 output_tuple, shape_index, &suballocation, in DoWork()
270 output_tuple->Unref(); in DoWork()
276 TF_RETURN_IF_ERROR(output_tuple->Intern(rm, &key)); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 51 xla::XlaOp output_tuple = TopK(context->Input(0), k); in Compile() local
52 context->SetOutput(0, xla::GetTupleElement(output_tuple, 0)); in Compile()
53 context->SetOutput(1, xla::GetTupleElement(output_tuple, 1)); in Compile()
|
| D | image_ops.cc | 539 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile() local
540 xla::XlaOp selected_indices_sorted = xla::GetTupleElement(output_tuple, 1); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 142 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormForwardTrainingThunk() argument
152 output_tuple_(output_tuple) { in CudnnBatchNormForwardTrainingThunk()
222 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo) in CudnnBatchNormBackwardThunk() argument
234 output_tuple_(output_tuple) { in CudnnBatchNormBackwardThunk()
|
| D | cudnn_batchnorm_thunk.h | 87 const BufferAllocation::Slice& output_tuple, const HloInstruction* hlo);
121 const BufferAllocation::Slice& output_tuple,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 298 HloInstruction* output_tuple = in TEST_F() local
306 EXPECT_EQ(computation->root_instruction(), output_tuple); in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | padding_fifo_queue_test.py | 1523 output_tuple = self.evaluate(output_tuple_t)
1525 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| D | fifo_queue_test.py | 1409 output_tuple = self.evaluate(output_tuple_t)
1411 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|