Searched refs:output_tuple (Results 1 – 10 of 10) sorted by relevance
| /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 | 545 xla::XlaOp output_tuple = TopK(scores_included, output_size); in Compile() local
546 xla::XlaOp selected_indices_sorted = xla::GetTupleElement(output_tuple, 1); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_util.cc | 299 const RefPtr<XRTTupleAllocation>& output_tuple, in RebuildOutputAliases() argument
306 return output_tuple->AliasBufferFrom(*input_tuples[alias.parameter_number], in RebuildOutputAliases()
347 RefPtr<XRTTupleAllocation> output_tuple, in CreateExecuteOutput() argument
349 if (return_exploded_tuple && output_tuple->on_host_shape().IsTuple()) { in CreateExecuteOutput()
351 xla::ShapeUtil::TupleElementCount(output_tuple->on_device_shape()); in CreateExecuteOutput()
359 output_tuple.get(), {i}, &suballocation, in CreateExecuteOutput()
368 memory_manager->Register(std::move(output_tuple)); in CreateExecuteOutput()
|
| D | xrt_util.h | 84 const RefPtr<XRTTupleAllocation>& output_tuple,
98 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()
288 *shaped_buffer, backend, device_ordinal, &output_tuple)); in CreateOutputTuple()
293 return RefPtr<XRTTupleAllocation>(output_tuple); in CreateOutputTuple()
488 RefPtr<XRTTupleAllocation> output_tuple, in DoWork()
494 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()
158 RefPtr<XRTTupleAllocation> output_tuple_ptr(output_tuple); in AllocateOutputTuple()
179 RefPtr<XRTTupleAllocation> output_tuple, in AllocateOutputTensors()
183 return CreateExecuteOutput(context, memory_manager, std::move(output_tuple), in AllocateOutputTensors()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | custom_call.md | 195 subbuffers of `output_tuple` are accessible by dereferencing `out`.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bfloat16_propagation_test.cc | 299 HloInstruction* output_tuple = in TEST_F() local
307 EXPECT_EQ(computation->root_instruction(), output_tuple); in TEST_F()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | padding_fifo_queue_test.py | 1589 output_tuple = self.evaluate(output_tuple_t)
1591 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|
| D | fifo_queue_test.py | 612 output_tuple = self.evaluate(output_tuple_t)
614 for (input_elem, output_elem) in zip(input_tuple, output_tuple):
|