| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 46 Outfeed(loop_counter, int_shape, ""); in XLA_TEST_F()
60 Outfeed(sum, xfeed_shape, ""); in XLA_TEST_F()
130 Outfeed(param, result_shape, ""); in XLA_TEST_F()
|
| D | local_client_execute_test.cc | 914 Outfeed(sum, shape, /*outfeed_config=*/""); in XLA_TEST_F()
|
| D | BUILD | 2572 # Outfeed ops are not supported on the interpreter backend.
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | outfeed_ops.cc | 42 xla::Outfeed(ctx->Input(0), xla_shape, outfeed_config); in Compile()
79 xla::Outfeed(tuple, tuple_shape, outfeed_config); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 101 TEST_F(XlaBuilderTest, Outfeed) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.cc | 1026 struct Outfeed { struct
1029 static Outfeed Handler() { return Outfeed(); } in Handler() argument
1033 Error Outfeed::operator()(const ServiceExecutableRunOptions* run_options, in operator ()()
1089 static bool Outfeed(runtime::KernelContext* ctx, void** args, void** attrs) { in Outfeed() function
1094 .To<RuntimeChecks()>(Outfeed::Handler()) in Outfeed()
2195 lib.Insert("xla.gpu.outfeed", &xla::gpu::Outfeed); in JitRtGpuCustomCalls()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_invariant_code_motion_test.cc | 294 Contains(op::Outfeed())); in TEST_F()
348 Contains(op::Outfeed())); in TEST_F()
|
| D | hlo_matchers.h | 286 HLO_MATCHER(Outfeed);
|
| D | hlo.proto | 169 // Outfeed configuration information, only present for kOutfeed.
|
| D | pattern_matcher.h | 2260 XLA_BINOP_PATTERN(Outfeed)
|
| D | copy_insertion_test.cc | 2364 EXPECT_THAT(outfeed, op::Outfeed(op::While(op::Copy(op::Parameter(0))), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | xla_service.proto | 102 // Transferred literal from the Outfeed buffer of the device.
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 386 HloModule Outfeed
400 HloModule Outfeed
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening_test.cc | 291 TEST_F(HloControlFlowFlatteningTest, Outfeed) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 645 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
1255 friend void Outfeed(XlaOp operand, const Shape& shape_with_layout,
2104 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
|
| D | xla_builder.cc | 1854 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed() function in xla::XlaBuilder
4410 void Outfeed(const XlaOp operand, const Shape& shape_with_layout, in Outfeed() function
4412 return operand.builder()->Outfeed(operand, shape_with_layout, outfeed_config); in Outfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 719 op::Outfeed(op::GetTupleElement(op::Parameter(), 0), in TEST_F()
740 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
760 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
785 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
815 op::Outfeed(first_outfeed, op::GetTupleElement()))); in TEST_F()
820 op::Outfeed(second_outfeed, op::GetTupleElement()))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.td | 723 // OutfeedOp corresponds to 'OutfeedWithToken' xla client API and not 'Outfeed'.
727 let summary = "Outfeed operator";
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 751 // OutfeedOp corresponds to 'OutfeedWithToken' xla client API and not 'Outfeed'.
755 let summary = "Outfeed operator";
|