| /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 | 122 xfeed->infeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue()
146 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 | 133 xfeed_manager->infeed()->EnqueueBuffersAtomically(buffers); in TransferLiteralToInfeed()
147 xfeed_manager->infeed()->EnqueueBuffersAtomically({buffer}); in TransferBufferToInfeed()
|
| D | ir_emitter.cc | 326 HloInfeedInstruction* infeed = Cast<HloInfeedInstruction>(instruction); in HandleInfeed() local
327 VLOG(2) << "HandleInfeed: " << infeed->ToString(); in HandleInfeed()
331 const Shape& data_shape = infeed->infeed_shape(); in HandleInfeed()
333 ShapeUtil::GetTupleElementShape(infeed->shape(), 0))); in HandleInfeed()
334 TF_RETURN_IF_ERROR(EmitTargetAddressForOp(infeed)); in HandleInfeed()
338 assignment_.GetUniqueSlice(infeed, {0})); in HandleInfeed()
341 assignment_.GetUniqueSlice(infeed, {1})); in HandleInfeed()
343 token_slice, ShapeUtil::GetTupleElementShape(infeed->shape(), 1)); in HandleInfeed()
344 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()
356 assignment_.GetUniqueSlice(infeed, {0, i})); in HandleInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 138 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local
139 auto addend = Reduce(infeed, ConstantR0<float>(&builder, 0.0f), in TEST_F()
227 auto infeed = Infeed(&builder, infeed_shape); in TEST_F() local
229 Reduce(GetTupleElement(infeed, 0), ConstantR0<float>(&builder, 0.0f), in TEST_F()
232 Tuple(&builder, {result, GetTupleElement(infeed, 1)}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 29 XlaOp infeed = InfeedWithToken(CreateToken(&b), scalar_s32); in BuildComputation() local
31 OutfeedWithToken(GetTupleElement(infeed, 0) + in BuildComputation()
33 GetTupleElement(infeed, 1), scalar_s32, "")); in BuildComputation()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_InfeedDequeueTuple.pbtxt | 7 A list of tensors that will be provided using the infeed mechanism.
22 summary: "Fetches multiple values from infeed as an XLA tuple."
|
| D | api_def_InfeedEnqueue.pbtxt | 7 A tensor that will be provided using the infeed mechanism.
27 be computed by the infeed operation.
|
| D | api_def_PrelinearizeTuple.pbtxt | 7 A list of tensors that will be provided using the infeed mechanism.
28 will be computed by the infeed operation.
|
| D | api_def_InfeedEnqueueTuple.pbtxt | 7 A list of tensors that will be provided using the infeed mechanism.
28 corresponding layout will be computed by the infeed operation.
|
| D | api_def_InfeedDequeue.pbtxt | 7 A tensor that will be provided using the infeed mechanism.
|
| D | api_def_InfeedEnqueuePrelinearizedBuffer.pbtxt | 17 summary: "An op which enqueues prelinearized buffer into TPU infeed."
|
| D | api_def_Prelinearize.pbtxt | 27 the infeed operation.
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_test.py | 76 infeed = tpu_feed.InfeedQueue(
81 return training_loop.repeat(5, training_step, infeed_queue=infeed)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 1065 infeed = ((u32[3]{0}, pred[]), token[]) infeed(token0) in CreateTestCases()
1066 infeed.data = (u32[3]{0}, pred[]) get-tuple-element(infeed), index=0 in CreateTestCases()
1067 outfeed = token[] outfeed(infeed.data, token0) in CreateTestCases()
1068 ROOT infeed.1 = ((u32[3]{0}, pred[]), token[]) infeed(token0) in CreateTestCases()
1069 infeed.1.data = (u32[3]{0}, pred[]) get-tuple-element(infeed.1), index=0 in CreateTestCases()
1070 infeed.1.token = token[] get-tuple-element(infeed.1), index=1 in CreateTestCases()
1071 outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token) in CreateTestCases()
|
| D | dfs_hlo_visitor_with_default.h | 115 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() argument
116 return DefaultAction(infeed); in HandleInfeed()
|
| D | hlo_runner.cc | 319 int64 num_threads = (options.infeed != nullptr) ? options.num_replicas : 0; in ExecuteReplicated()
328 if (options.infeed != nullptr) { in ExecuteReplicated()
338 executor, *options.infeed)); in ExecuteReplicated()
|
| D | hlo_runner.h | 58 const Literal* infeed = nullptr; member
|
| D | hlo_dce_test.cc | 238 auto infeed = body_builder.AddInstruction( in TEST_F() local
241 HloInstruction::CreateBinary(shape, HloOpcode::kAdd, param, infeed)); in TEST_F()
|
| D | hlo_cost_analysis.h | 80 Status HandleInfeed(const HloInstruction* infeed) override;
|
| D | hlo_evaluator.h | 259 Status HandleInfeed(HloInstruction* infeed) override { in HandleInfeed() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 125 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
126 LOG(ERROR) << " " << ShapeUtil::HumanString(Shape(infeed.shape())) << " " in GetXfeedShape()
127 << infeed.name(); in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 85 Status HandleInfeed(HloInstruction* infeed) override;
|