| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | topk_op.cc | 67 xla::XlaOp output_tuple = TopK(context->Input(0), k); in Compile() local
68 auto values = xla::GetTupleElement(output_tuple, 0); in Compile()
69 auto indices = xla::GetTupleElement(output_tuple, 1); in Compile()
|
| D | approx_topk_op.cc | 88 xla::XlaOp output_tuple = ApproxTopKFn( in Compile() local
92 ctx->SetOutput(0, xla::GetTupleElement(output_tuple, 0)); in Compile()
93 ctx->SetOutput(1, xla::GetTupleElement(output_tuple, 1)); in Compile()
|
| D | image_ops.cc | 569 xla::XlaOp output_tuple = TopK(scores_included, output_size); in ComputeResult() local
570 xla::XlaOp selected_indices_sorted = xla::GetTupleElement(output_tuple, 1); in ComputeResult()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 302 const RefPtr<XRTTupleAllocation>& output_tuple, in RebuildOutputAliases() argument
309 return output_tuple->AliasBufferFrom(*input_tuples[alias.parameter_number], in RebuildOutputAliases()
350 RefPtr<XRTTupleAllocation> output_tuple, in CreateExecuteOutput() argument
352 if (return_exploded_tuple && output_tuple->on_host_shape().IsTuple()) { in CreateExecuteOutput()
354 xla::ShapeUtil::TupleElementCount(output_tuple->on_device_shape()); in CreateExecuteOutput()
362 output_tuple.get(), {i}, &suballocation, in CreateExecuteOutput()
372 memory_manager->Register(std::move(output_tuple)); in CreateExecuteOutput()
|
| D | xrt_util.h | 85 const RefPtr<XRTTupleAllocation>& output_tuple,
99 RefPtr<XRTTupleAllocation> output_tuple,
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 270 XRTTupleAllocation* output_tuple; in CreateOutputTuple() local
284 output_device_shape, backend, device_ordinal, &output_tuple, in CreateOutputTuple()
289 *shaped_buffer, backend, device_ordinal, &output_tuple, allocator)); in CreateOutputTuple()
294 return RefPtr<XRTTupleAllocation>(output_tuple); in CreateOutputTuple()
492 RefPtr<XRTTupleAllocation> output_tuple, in DoWork()
498 std::move(output_tuple), in DoWork()
|
| D | tpu_execute_op.cc | 151 XRTTupleAllocation* output_tuple; in AllocateOutputTuple() local
157 node_context->backend(), device_ordinal, &output_tuple, in AllocateOutputTuple()
159 RefPtr<XRTTupleAllocation> output_tuple_ptr(output_tuple); in AllocateOutputTuple()
180 RefPtr<XRTTupleAllocation> output_tuple, in AllocateOutputTensors()
184 return CreateExecuteOutput(context, memory_manager, std::move(output_tuple), in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 300 HloInstruction* output_tuple = in TEST_F() local
308 EXPECT_EQ(computation->root_instruction(), output_tuple); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | custom_call.md | 258 subbuffers of `output_tuple` are accessible by dereferencing `out`.
|
| /external/tensorflow/tensorflow/python/kernel_tests/data_structures/ |
| D | padding_fifo_queue_test.py | 1584 output_tuple = self.evaluate(output_tuple_t)
1586 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| D | fifo_queue_test.py | 607 output_tuple = self.evaluate(output_tuple_t)
609 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 5662 auto output_tuple = op::Tuple( in TEST_F() local
5683 output_tuple, op::GetTupleElement(op::Parameter(0)), next_i)); in TEST_F()
5766 auto output_tuple = op::Tuple( in TEST_F() local
5785 output_tuple, op::GetTupleElement(op::Parameter(0)), next_i)); in TEST_F()
5881 auto output_tuple = in TEST_F() local
5892 output_tuple, in TEST_F()
6011 auto output_tuple = op::Tuple( in TEST_F() local
6028 output_tuple, op::GetTupleElement(op::Parameter(0)), next_i)); in TEST_F()
6121 auto output_tuple = op::Tuple( in TEST_F() local
6131 output_tuple, in TEST_F()
|