| /external/tensorflow/tensorflow/core/profiler/ |
| D | profiler_service_monitor_result.proto | 14 // Device utilization, device idle time, step time, and infeed percentage
31 // Average infeed percentage.
33 // Minimum infeed percentage.
35 // Maximum infeed percentage.
|
| /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 | 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()
|
| /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/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_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_InfeedEnqueue.pbtxt | 7 A tensor that will be provided using the infeed mechanism.
27 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/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/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 83 auto infeed = InfeedWithToken(token, ShapeUtil::MakeShape(F32, {4, 8}), ""); in TEST_F() local
87 GetMlirOpString(infeed), in TEST_F()
88 …R"("mhlo.infeed"(%0) {infeed_config = ""} : (!mhlo.token) -> tuple<tensor<4x8xf32>, !mhlo.token>)"… in TEST_F()
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | op_metrics.proto | 92 // The total host infeed-enqueue duration in picoseconds.
95 // consecutive infeed-enqueues (per host) in picoseconds.
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 1224 infeed = ((u32[3]{0}, pred[]), token[]) infeed(token0) in CreateTestCases()
1225 infeed.data = (u32[3]{0}, pred[]) get-tuple-element(infeed), index=0 in CreateTestCases()
1226 outfeed = token[] outfeed(infeed.data, token0), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
1227 ROOT infeed.1 = ((u32[3]{0}, pred[]), token[]) infeed(token0) in CreateTestCases()
1228 infeed.1.data = (u32[3]{0}, pred[]) get-tuple-element(infeed.1), index=0 in CreateTestCases()
1229 infeed.1.token = token[] get-tuple-element(infeed.1), index=1 in CreateTestCases()
1230 outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
|
| D | dfs_hlo_visitor_with_default.h | 134 Status HandleInfeed(HloInstructionPtr infeed) override { in HandleInfeed() argument
135 return DefaultAction(infeed); in HandleInfeed()
|
| D | hlo_runner.cc | 252 int64 num_threads = (options.infeed != nullptr) ? options.num_replicas : 0; in ExecuteReplicatedImpl()
261 if (options.infeed != nullptr) { in ExecuteReplicatedImpl()
271 executor, *options.infeed)); in ExecuteReplicatedImpl()
|
| D | hlo_cost_analysis_test.cc | 814 HloInstruction* infeed = in TEST_F() local
820 ASSERT_IS_OK(infeed->Accept(&analysis)); in TEST_F()
823 EXPECT_EQ(analysis.bytes_accessed(*infeed), sizeof(float) * 2 * 3); in TEST_F()
824 EXPECT_EQ(analysis.operand_bytes_accessed(*infeed, 0), 0); in TEST_F()
825 EXPECT_EQ(analysis.output_bytes_accessed(*infeed), sizeof(float) * 2 * 3); in TEST_F()
|
| D | hlo_runner_interface.h | 55 const Literal* infeed = nullptr; member
|
| D | hlo_dce_test.cc | 276 auto infeed = body_builder.AddInstruction( in TEST_F() local
279 HloInstruction::CreateGetTupleElement(shape, infeed, 0)); in TEST_F()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_test.py | 104 infeed = tpu_feed.InfeedQueue(
109 return training_loop.repeat(5, training_step, infeed_queue=infeed)
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 150 for (const auto& infeed : xfeed_instrs) { in GetXfeedShape() local
151 LOG(ERROR) << " " << ShapeUtil::HumanString(Shape(infeed.shape())) << " " in GetXfeedShape()
152 << infeed.name(); in GetXfeedShape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 84 Status HandleInfeed(HloInstruction* infeed) override;
|