Searched refs:Infeed (Results 1 – 10 of 10) sorted by relevance
317 ComputationDataHandle LocalComputationBuilder::Infeed(const Shape& shape) { in Infeed() function in xla::swig::LocalComputationBuilder318 return builder_.Infeed(shape); in Infeed()
144 ComputationDataHandle Infeed(const Shape& shape);
473 def Infeed(self, shape): member in ComputationBuilder482 return _wrap_data_handle(self._client.Infeed(shape))
1263 c.Infeed(xla_client.Shape.from_numpy(to_infeed[0]))1275 x = c.Infeed(xla_client.Shape.from_numpy(to_round_trip[0]))
879 %unignore xla::swig::LocalComputationBuilder::Infeed;
125 HLO_MATCHER(Infeed);
1240 ## Infeed section in Operation Semantics1243 [`ComputationBuilder::Infeed`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/compu…1245 <b> `Infeed(shape)` </b>1249 | `shape` | `Shape` | Shape of the data read from the Infeed interface. The |1254 Reads a single data item from the implicit Infeed streaming interface of the1256 `ComputationDataHandle` of the data. Multiple Infeed operations are allowed in a1257 computation, but there must be a total order among the Infeed operations. For1263 Infeed(shape)1267 Infeed(shape)1271 Nested tuple shapes are not supported. For an empty tuple shape, the Infeed[all …]
424 ComputationDataHandle Infeed(const Shape& shape, const string& config = "");
746 ComputationDataHandle ComputationBuilder::Infeed(const Shape& shape, in Infeed() function in xla::ComputationBuilder
874 auto in = builder.Infeed(shape); in XLA_TEST_F()