/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | xfeed_manager_test.cc | 90 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F() 91 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F() 102 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F() 104 xfeed->infeed()->EnqueueBuffersAtomically({b}); in TEST_F() 125 xfeed->infeed()->EnqueueBuffersAtomically({a}); in TEST_F()
|
D | cpu_runtime.cc | 90 xfeed->infeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() 109 xfeed->infeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
|
D | xfeed_manager.h | 113 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
|
D | xfeed_manager.cc | 26 infeed()->Reset(); in Reset()
|
D | cpu_transfer_manager.cc | 132 xfeed_manager->infeed()->EnqueueBuffersAtomically(buffers); in TransferLiteralToInfeed() 145 xfeed_manager->infeed()->EnqueueBuffersAtomically({buffer}); in TransferBufferToInfeed()
|
D | ir_emitter.cc | 330 Status IrEmitter::HandleInfeed(HloInstruction* infeed) { in HandleInfeed() argument 331 VLOG(2) << "HandleInfeed: " << infeed->ToString(); in HandleInfeed() 333 const Shape& shape = infeed->shape(); in HandleInfeed() 337 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(infeed)); in HandleInfeed() 338 llvm_ir::IrArray infeed_array = GetIrArrayFor(infeed); in HandleInfeed() 350 assignment_.GetUniqueSlice(infeed, {i})); in HandleInfeed() 372 GetEmittedValueFor(infeed))); in HandleInfeed()
|
D | ir_emitter.h | 129 Status HandleInfeed(HloInstruction* infeed) override;
|
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
D | tpu_test.py | 68 infeed = tpu_feed.InfeedQueue( 73 return training_loop.repeat(5, training_step, infeed_queue=infeed)
|
/external/tensorflow/tensorflow/contrib/tpu/profiler/ |
D | tf_op_stats.proto | 64 // The total host infeed-enqueue duration in picoseconds. 67 // consecutive infeed-enqueues (per host) in picoseconds. 77 // The infeed duration in picoseconds.
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | dfs_hlo_visitor_with_default.h | 100 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() argument 101 return DefaultAction(infeed); in HandleInfeed()
|
D | hlo_dce_test.cc | 238 auto infeed = in TEST_F() local 241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed)); in TEST_F()
|
D | hlo_cost_analysis.h | 72 Status HandleInfeed(const HloInstruction* infeed) override;
|
D | layout_assignment_test.cc | 698 auto infeed = in TEST_F() local 700 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed})); in TEST_F()
|
D | hlo.proto | 115 // The string representation of the infeed configuration.
|
D | buffer_assignment_test.cc | 1637 auto infeed = builder.AddInstruction(HloInstruction::CreateInfeed(r0s32, "")); in TEST_F() local 1641 HloInstruction::CreateWhile(r0s32, cond0, body0, infeed)); in TEST_F() 1672 sequence[module->entry_computation()] = {infeed, while0, while1, zero, in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
D | hlo_parser_test.cc | 772 infeed = (u32[3]{0}, pred[]) infeed() 773 outfeed = () outfeed(infeed) 774 ROOT infeed.1 = (u32[3]{0}, pred[]) infeed() 775 outfeed.1 = () outfeed(infeed.1)
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.h | 80 Status HandleInfeed(HloInstruction* infeed) override;
|
D | ir_emitter_unnested.cc | 1960 Status IrEmitterUnnested::HandleInfeed(HloInstruction* infeed) { in HandleInfeed() argument 1961 thunk_sequence_->emplace_back(BuildInfeedThunk(infeed)); in HandleInfeed()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | xla_data.proto | 520 // The shape of the data returned by reading the device's infeed buffer. 523 // Additional infeed configuration for the backend.
|
/external/tensorflow/tensorflow/docs_src/performance/xla/ |
D | operation_semantics.md | 597 is possible when a while loop's condition depends on data from infeed and the
|