| /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 | 929 Outfeed(sum, shape, /*outfeed_config=*/""); in XLA_TEST_F()
|
| D | BUILD | 2405 # 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 | 91 TEST_F(XlaBuilderTest, Outfeed) { in TEST_F() argument
|
| /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 | 247 HLO_MATCHER(Outfeed);
|
| D | hlo.proto | 114 // Outfeed configuration information, only present for kOutfeed.
|
| D | pattern_matcher.h | 2087 XLA_BINOP_PATTERN(Outfeed)
|
| /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 | 468 HloModule Outfeed
482 HloModule Outfeed
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 421 // OutfeedOp corresponds to 'OutfeedWithToken' xla client API and not 'Outfeed'.
425 string summary = "Outfeed operator";
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 641 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
1211 friend void Outfeed(XlaOp operand, const Shape& shape_with_layout,
2002 void Outfeed(XlaOp operand, const Shape& shape_with_layout,
|
| D | xla_builder.cc | 1768 void XlaBuilder::Outfeed(XlaOp operand, const Shape& shape_with_layout, in Outfeed() function in xla::XlaBuilder
4453 void Outfeed(const XlaOp operand, const Shape& shape_with_layout, in Outfeed() function
4455 return operand.builder()->Outfeed(operand, shape_with_layout, outfeed_config); in Outfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 684 op::Outfeed(op::GetTupleElement(op::Parameter(), 0), in TEST_F()
705 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
725 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
750 op::Outfeed(op::Parameter(), op::AfterAll()))); in TEST_F()
780 op::Outfeed(first_outfeed, op::GetTupleElement()))); in TEST_F()
785 op::Outfeed(second_outfeed, op::GetTupleElement()))); in TEST_F()
|