| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_OutfeedDequeueV2.pbtxt | 15 A tensor that will be read from the device outfeed.
31 Retrieves a single tensor from the computation outfeed. Device ordinal is a
32 tensor allowing dynamic outfeed.
|
| D | api_def_OutfeedDequeueTupleV2.pbtxt | 15 A list of tensors that will be read from the outfeed.
31 Retrieve multiple values from the computation outfeed. Device ordinal is a
32 tensor allowing dynamic outfeed.
|
| 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 | 95 Literal outfeed(ShapeUtil::MakeShape(S32, {})); in TestWithDeviceCount() local
96 TF_ASSERT_OK(client->TransferFromOutfeedLocal(device_ordinal, &outfeed)); in TestWithDeviceCount()
97 EXPECT_EQ(outfeed, LiteralUtil::CreateR0<int32>(device_ordinal * 100 + 1)); in TestWithDeviceCount()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | xfeed_manager_test.cc | 134 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()
142 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()
150 xfeed->outfeed()->EnqueueBuffersAtomically({b}); in TEST_F()
|
| D | xfeed_manager.h | 114 XfeedQueueManager* outfeed() { return &outfeed_; } in outfeed() function
|
| D | xfeed_manager.cc | 27 outfeed()->Reset(); in Reset()
|
| D | cpu_transfer_manager.cc | 272 xfeed_manager->outfeed()->EnqueueBuffersAtomically(buffer_pointers); in TransferBuffersFromOutfeedInternal()
|
| D | cpu_runtime.cc | 289 xfeed->outfeed()->BlockingDequeueBuffer(); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation()
313 xfeed->outfeed()->ReleaseCurrentBuffer(buffer_length, buffer_ptr, in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation()
|
| D | ir_emitter.h | 145 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 97 auto outfeed = OutfeedWithToken(data, token, outfeed_shape, ""); in TEST_F() local
101 GetMlirOpString(outfeed), in TEST_F()
102 …R"("mhlo.outfeed"(%0, %1) {outfeed_config = ""} : (tensor<4x8xf32>, !mhlo.token) -> !mhlo.token)"); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 137 Status HandleOutfeed(HloInstructionPtr outfeed) override { in HandleOutfeed() argument
138 return DefaultAction(outfeed); in HandleOutfeed()
|
| D | hlo_cost_analysis_test.cc | 816 HloInstruction* outfeed = in TEST_F() local
821 ASSERT_IS_OK(outfeed->Accept(&analysis)); in TEST_F()
827 EXPECT_EQ(analysis.bytes_accessed(*outfeed), sizeof(float) * 2 * 3); in TEST_F()
828 EXPECT_EQ(analysis.operand_bytes_accessed(*outfeed, 0), in TEST_F()
830 EXPECT_EQ(analysis.output_bytes_accessed(*outfeed), 0); in TEST_F()
|
| D | hlo_verifier.cc | 432 HloOutfeedInstruction* outfeed = Cast<HloOutfeedInstruction>(instruction); in HandleOutfeed() local
437 if (!ShapesSame(outfeed->outfeed_shape(), outfeed->operand(0)->shape())) { in HandleOutfeed()
441 StringifyShape(outfeed->operand(0)->shape()), in HandleOutfeed()
442 StringifyShape(outfeed->outfeed_shape()), outfeed->ToString()); in HandleOutfeed()
444 return CheckShape(outfeed, ShapeUtil::MakeTokenShape()); in HandleOutfeed()
|
| D | hlo_evaluator.h | 286 Status HandleOutfeed(HloInstruction* outfeed) override { in HandleOutfeed() argument
|
| D | hlo_cost_analysis.h | 88 Status HandleOutfeed(const HloInstruction* outfeed) override;
|
| D | hlo_cost_analysis.cc | 352 Status HloCostAnalysis::HandleOutfeed(const HloInstruction* outfeed) { in HandleOutfeed() argument
355 for (int64 i = 0; i < outfeed->operand_count(); ++i) { in HandleOutfeed()
356 const HloInstruction* operand = outfeed->operand(i); in HandleOutfeed()
|
| D | hlo_parser_test.cc | 1226 outfeed = token[] outfeed(infeed.data, token0), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
1230 outfeed.1 = token[] outfeed(infeed.1.data, infeed.1.token), outfeed_shape=(u32[3]{0}, pred[]) in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 85 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| D | ir_emitter_unnested.h | 197 Status HandleOutfeed(HloInstruction* outfeed) override;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 285 Literal outfeed(*outfeed_shape); in ReplayComputation() local
287 client->TransferFromOutfeedLocal(/*device_ordinal=*/0, &outfeed)); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 471 // CHECK: "lmhlo.outfeed"
477 ROOT %o = token[] outfeed(f32[3] %source, token[] %tok)
487 // CHECK: "lmhlo.outfeed"
493 ROOT %o = token[] outfeed((f32[3], f16[5]) %tuple, token[] %tok)
|