| /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 | 877 auto in = Infeed(&builder, shape); in XLA_TEST_F()
903 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/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 44 Infeed(&builder, literal.shape()); in TestInfeedRoundTrip()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops.td | 350 // InfeedOp corresponds to 'InfeedWithToken' xla client API and not 'Infeed'.
354 string summary = "Infeed operator";
357 Reads a single data item from the implicit Infeed streaming interface of
359 of the data. Multiple Infeed operations are allowed in a computation, but
360 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 | 1651 ## Infeed section in Operation Semantics
1654 [`XlaBuilder::Infeed`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builder.h…
1656 <b> `Infeed(shape)` </b>
1660 | `shape` | `Shape` | Shape of the data read from the Infeed interface. The |
1665 Reads a single data item from the implicit Infeed streaming interface of the
1667 `XlaOp` of the data. Multiple Infeed operations are allowed in a
1668 computation, but there must be a total order among the Infeed operations. For
1674 Infeed(shape)
1678 Infeed(shape)
1682 Nested tuple shapes are not supported. For an empty tuple shape, the Infeed
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 234 HLO_MATCHER(Infeed);
|
| D | pattern_matcher.h | 2101 XLA_UNOP_PATTERN(Infeed)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 468 XlaOp Infeed(const Shape& shape, const string& config = "");
857 friend XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
1481 XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
|
| D | xla_builder.cc | 1340 XlaOp XlaBuilder::Infeed(const Shape& shape, const string& config) { in Infeed() function in xla::XlaBuilder
3132 XlaOp Infeed(XlaBuilder* builder, const Shape& shape, const string& config) { in Infeed() function
3133 return builder->Infeed(shape, config); in Infeed()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1924 c.GetTupleElement(c.Infeed(xla_client.shape_from_pyval(to_infeed[0])), 0)
1936 c.GetTupleElement(c.Infeed(xla_client.shape_from_pyval(to_infeed)), 0)
1947 x_and_token = c.Infeed(xla_client.shape_from_pyval(to_round_trip[0]))
|
| D | xla_client.py | 780 def Infeed(self, shape, token=None): member in ComputationBuilder
|