Searched defs:BroadcastedIota (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | local_computation_builder.cc | 465 LocalOp ComputationBuilder::BroadcastedIota(const Shape& shape, in BroadcastedIota() function in xla::swig::ComputationBuilder |
| D | xla_client.py | 1078 def BroadcastedIota(self, dtype, shape, dimension): member in ComputationBuilder |