| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 38 XlaOp num_iter = Infeed(&b, int_shape); in XLA_TEST_F()
58 XlaOp some_input = Infeed(&body_builder, xfeed_shape); in XLA_TEST_F()
141 XlaOp pred = Infeed(&b, condition_shape); in XLA_TEST_F()
|
| D | local_client_execute_test.cc | 900 auto in = Infeed(&builder, shape); in XLA_TEST_F()
926 auto in = Infeed(&builder, shape); in XLA_TEST_F()
|
| D | while_test.cc | 1240 Infeed(&builder, ShapeUtil::MakeShape(PRED, {})); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 50 Infeed(&builder, literal.shape()); in TestInfeedRoundTrip()
138 auto infeed = Infeed(&builder, infeed_shape); in TEST_F()
227 auto infeed = Infeed(&builder, infeed_shape); in TEST_F()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | infeed_op.cc | 99 ctx->SetOutput(0, xla::Infeed(b, xla_shape_)); in Compile()
140 auto tuple = xla::Infeed(b, tuple_shape_); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 44 Infeed(&builder, literal.shape()); in TestInfeedRoundTrip()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | mlir_hlo_builder_test.cc | 81 TEST_F(XlaBuilderTest, Infeed) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 392 // InfeedOp corresponds to 'InfeedWithToken' xla client API and not 'Infeed'.
396 string summary = "Infeed operator";
399 Reads a single data item from the implicit Infeed streaming interface of
401 of the data. Multiple Infeed operations are allowed in a computation, but
402 there must be a total order among the Infeed operations.
|
| D | hlo_ops_base.td | 1318 string summary = "Infeed operator";
1321 Reads a single data item from the implicit Infeed streaming interface of
1323 returns an LHLO op of the data. Multiple Infeed operations are allowed in a
1324 computation, but there must be a total order among the Infeed operations.
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | xla_service.proto | 97 // Transfers the given literal to the Infeed buffer of the device.
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1721 ## Infeed section in Operation Semantics
1724 [`XlaBuilder::Infeed`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builder.h…
1726 <b> `Infeed(shape)` </b>
1730 | `shape` | `Shape` | Shape of the data read from the Infeed interface. The |
1735 Reads a single data item from the implicit Infeed streaming interface of the
1737 `XlaOp` of the data. Multiple Infeed operations are allowed in a
1738 computation, but there must be a total order among the Infeed operations. For
1744 Infeed(shape)
1748 Infeed(shape)
1752 Nested tuple shapes are not supported. For an empty tuple shape, the Infeed
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 236 HLO_MATCHER(Infeed);
|
| D | pattern_matcher.h | 2019 XLA_UNOP_PATTERN(Infeed)
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 314 AllOf(op::Infeed(), op::Shape("(f32[4,2]{1,0}, token[])"))), in TEST_F()
340 AllOf(op::Shape("(f32[5,2], token[])"), op::Infeed(op::Parameter()))); in TEST_F()
342 AllOf(op::Shape("(f32[4,2], token[])"), op::Infeed(op::Parameter())); in TEST_F()
371 op::Infeed(op::Parameter()))); in TEST_F()
373 op::Infeed(op::Parameter())); in TEST_F()
405 op::Infeed(op::Parameter())); in TEST_F()
413 AllOf(op::Shape("(((), f32[2]), token[])"), op::Infeed(op::Parameter())); in TEST_F()
1033 op::GetTupleElement(op::Infeed())); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 636 XlaOp Infeed(const Shape& shape, const string& config = "");
1209 friend XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
1986 XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
|
| D | xla_builder.cc | 1663 XlaOp XlaBuilder::Infeed(const Shape& shape, const string& config) { in Infeed() function in xla::XlaBuilder
4449 XlaOp Infeed(XlaBuilder* builder, const Shape& shape, const string& config) { in Infeed() function
4450 return builder->Infeed(shape, config); in Infeed()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 456 HloModule Infeed
|