Searched defs:iota_dimension (Results 1 – 9 of 9) sorted by relevance
329 int64_t iota_dimension) { in MakeIotaHlo()
2217 int64_t iota_dimension() const { return iota_dimension_; } in iota_dimension() function
977 int64_t iota_dimension = iota->iota_dimension(); in HandleIota() local
1303 optional<int64_t> iota_dimension; in CreateInstruction() local
3242 int64_t iota_dimension) in HloIotaInstruction()
1019 const Shape& shape, int64_t iota_dimension) { in CreateIota()
272 XlaOp MlirHloBuilder::Iota(const Shape& shape, int64_t iota_dimension) { in Iota()
876 XlaOp XlaBuilder::Iota(const Shape& shape, int64_t iota_dimension) { in Iota()5000 XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64_t iota_dimension) { in Iota()
4372 IntegerAttr iota_dimension = in matchAndRewrite() local