| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_OutfeedEnqueue.pbtxt | 7 A tensor that will be inserted into the outfeed queue.
10 summary: "Enqueue a Tensor on the computation outfeed."
|
| D | api_def_OutfeedEnqueueTuple.pbtxt | 7 A list of tensors that will be inserted into the outfeed queue as an
11 summary: "Enqueue multiple Tensor values on the computation outfeed."
|
| D | api_def_OutfeedDequeue.pbtxt | 7 A tensor that will be read from the device outfeed.
30 summary: "Retrieves a single tensor from the computation outfeed."
|
| D | api_def_OutfeedDequeueTuple.pbtxt | 7 A list of tensors that will be read from the outfeed.
30 summary: "Retrieve multiple values from the computation outfeed."
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 93 TF_ASSERT_OK_AND_ASSIGN(Literal outfeed, in TestWithDeviceCount()
96 EXPECT_EQ(outfeed, LiteralUtil::CreateR0<int32>(device_ordinal * 100 + 1)); in TestWithDeviceCount()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 165 xfeed->outfeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
189 xfeed->outfeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
|
| D | xfeed_manager.h | 114 XfeedQueueManager* outfeed() { return &outfeed_; } in outfeed() function
|
| D | xfeed_manager.cc | 27 outfeed()->Reset(); in Reset()
|
| D | xfeed_manager_test.cc | 134 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()
|
| D | cpu_transfer_manager.cc | 272 xfeed_manager->outfeed()->EnqueueBuffersAtomically(buffer_pointers); in TransferBuffersFromOutfeedInternal()
|
| D | ir_emitter.h | 159 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| D | ir_emitter.cc | 474 Status IrEmitter::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() argument
479 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(outfeed)); in HandleOutfeed()
481 HloInstruction* operand = outfeed->operands()[0]; in HandleOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 118 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed() argument
119 return DefaultAction(outfeed); in HandleOutfeed()
|
| D | hlo_verifier.cc | 257 HloOutfeedInstruction* outfeed = Cast<HloOutfeedInstruction>(instruction); in HandleOutfeed() local
262 if (!ShapesSame(outfeed->outfeed_shape(), outfeed->operand(0)->shape())) { in HandleOutfeed()
266 StringifyShape(outfeed->operand(0)->shape()), in HandleOutfeed()
267 StringifyShape(outfeed->outfeed_shape()), outfeed->ToString()); in HandleOutfeed()
269 return CheckShape(outfeed, ShapeUtil::MakeTokenShape()); in HandleOutfeed()
|
| D | hlo_cost_analysis.h | 81 Status HandleOutfeed(const HloInstruction* outfeed) override;
|
| D | hlo_evaluator.h | 262 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed() argument
|
| D | hlo_parser_test.cc | 1067 outfeed = token[] outfeed(infeed.data, token0) in CreateTestCases()
1071 outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token) in CreateTestCases()
|
| D | hlo.proto | 137 // Shape of outfeed request.
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 86 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| D | ir_emitter_unnested.h | 174 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| D | ir_emitter_unnested.cc | 1478 Status IrEmitterUnnested::HandleOutfeed(HloInstruction* outfeed) { in HandleOutfeed() argument
1479 AddThunkToThunkSequence(BuildOutfeedThunk(outfeed)); in HandleOutfeed()
|