| /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 | 849 auto in = Infeed(&builder, shape); in XLA_TEST_F()
875 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/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 | 1495 ## Infeed section in Operation Semantics
1498 [`XlaBuilder::Infeed`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builder.h…
1500 <b> `Infeed(shape)` </b>
1504 | `shape` | `Shape` | Shape of the data read from the Infeed interface. The |
1509 Reads a single data item from the implicit Infeed streaming interface of the
1511 `XlaOp` of the data. Multiple Infeed operations are allowed in a
1512 computation, but there must be a total order among the Infeed operations. For
1518 Infeed(shape)
1522 Infeed(shape)
1526 Nested tuple shapes are not supported. For an empty tuple shape, the Infeed
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 448 LocalOp ComputationBuilder::Infeed(const Shape& shape) { in Infeed() function in xla::swig::ComputationBuilder
449 return xla::Infeed(&builder_, shape); in Infeed()
|
| D | local_computation_builder.h | 229 LocalOp Infeed(const Shape& shape);
|
| D | local_computation_builder.i | 294 %unignore xla::swig::ComputationBuilder::Infeed;
|
| D | xla_client.py | 942 def Infeed(self, shape): member in ComputationBuilder
951 return self._client.Infeed(shape)
|
| D | xla_client_test.py | 1652 c.Infeed(xla_client.Shape.from_pyval(to_infeed[0]))
1664 x = c.Infeed(xla_client.Shape.from_pyval(to_round_trip[0]))
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 208 HLO_MATCHER(Infeed);
|
| D | pattern_matcher.h | 2088 XLA_UNOP_PATTERN(Infeed)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 422 XlaOp Infeed(const Shape& shape, const string& config = "");
818 friend XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
1393 XlaOp Infeed(XlaBuilder* builder, const Shape& shape,
|
| D | xla_builder.cc | 1281 XlaOp XlaBuilder::Infeed(const Shape& shape, const string& config) { in Infeed() function in xla::XlaBuilder
3067 XlaOp Infeed(XlaBuilder* builder, const Shape& shape, const string& config) { in Infeed() function
3068 return builder->Infeed(shape, config); in Infeed()
|