Searched defs:iota_dimension (Results 1 – 10 of 10) sorted by relevance
281 int64 iota_dimension) { in MakeIotaHlo()
1749 int64 iota_dimension() const { return iota_dimension_; } in iota_dimension() function
614 int64 iota_dimension = iota->iota_dimension(); in HandleIota() local
1070 optional<int64> iota_dimension; in ParseInstructionRhs() local
840 const Shape& shape, int64 iota_dimension) { in CreateIota()
2918 HloIotaInstruction::HloIotaInstruction(const Shape& shape, int64 iota_dimension) in HloIotaInstruction()
201 XlaOp MlirHloBuilder::Iota(const Shape& shape, int64 iota_dimension) { in Iota()
295 auto iota_dimension = op.iota_dimension(); in Verify() local314 auto iota_dimension = iota.iota_dimension(); in matchAndRewrite() local380 auto iota_dimension = iota.iota_dimension(); in matchAndRewrite() local
810 XlaOp XlaBuilder::Iota(const Shape& shape, int64 iota_dimension) { in Iota()4966 XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64 iota_dimension) { in Iota()
3864 IntegerAttr iota_dimension = in matchAndRewrite() local