Searched refs:BroadcastedIota (Results 1 – 5 of 5) sorted by relevance
238 LocalOp BroadcastedIota(const Shape& shape, int64 dimension);
299 %unignore xla::swig::ComputationBuilder::BroadcastedIota;
1078 def BroadcastedIota(self, dtype, shape, dimension): member in ComputationBuilder1090 return self._client.BroadcastedIota(xla_shape, dimension)
465 LocalOp ComputationBuilder::BroadcastedIota(const Shape& shape, in BroadcastedIota() function in xla::swig::ComputationBuilder
196 c.BroadcastedIota(np.int64, (2, 3), 1)