Home
last modified time | relevance | path

Searched refs:infeed (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dxfeed_manager_test.cc90 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()
Dcpu_runtime.cc90 xfeed->infeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
109 xfeed->infeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue()
Dxfeed_manager.h113 XfeedQueueManager* infeed() { return &infeed_; } in infeed() function
Dxfeed_manager.cc26 infeed()->Reset(); in Reset()
Dcpu_transfer_manager.cc132 xfeed_manager->infeed()->EnqueueBuffersAtomically(buffers); in TransferLiteralToInfeed()
145 xfeed_manager->infeed()->EnqueueBuffersAtomically({buffer}); in TransferBufferToInfeed()
Dir_emitter.cc330 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()
Dir_emitter.h129 Status HandleInfeed(HloInstruction* infeed) override;
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dtpu_test.py68 infeed = tpu_feed.InfeedQueue(
73 return training_loop.repeat(5, training_step, infeed_queue=infeed)
/external/tensorflow/tensorflow/contrib/tpu/profiler/
Dtf_op_stats.proto64 // 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/
Ddfs_hlo_visitor_with_default.h100 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() argument
101 return DefaultAction(infeed); in HandleInfeed()
Dhlo_dce_test.cc238 auto infeed = in TEST_F() local
241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed)); in TEST_F()
Dhlo_cost_analysis.h72 Status HandleInfeed(const HloInstruction* infeed) override;
Dlayout_assignment_test.cc698 auto infeed = in TEST_F() local
700 false_builder.AddInstruction(HloInstruction::CreateTuple({infeed})); in TEST_F()
Dhlo.proto115 // The string representation of the infeed configuration.
Dbuffer_assignment_test.cc1637 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/
Dhlo_parser_test.cc772 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/
Dir_emitter.h80 Status HandleInfeed(HloInstruction* infeed) override;
Dir_emitter_unnested.cc1960 Status IrEmitterUnnested::HandleInfeed(HloInstruction* infeed) { in HandleInfeed() argument
1961 thunk_sequence_->emplace_back(BuildInfeedThunk(infeed)); in HandleInfeed()
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto520 // 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/
Doperation_semantics.md597 is possible when a while loop's condition depends on data from infeed and the