Searched defs:iota_dimension (Results 1 – 8 of 8) sorted by relevance
259 int64 iota_dimension) { in MakeIotaHlo()
1543 int64 iota_dimension() const { return iota_dimension_; } in iota_dimension() function
509 int64 iota_dimension = iota->iota_dimension(); in HandleIota() local
748 optional<int64> iota_dimension; in ParseInstructionRhs() local
724 const Shape& shape, int64 iota_dimension) { in CreateIota()
2634 HloIotaInstruction::HloIotaInstruction(const Shape& shape, int64 iota_dimension) in HloIotaInstruction()
702 XlaOp XlaBuilder::Iota(const Shape& shape, int64 iota_dimension) { in Iota()3581 XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64 iota_dimension) { in Iota()
2208 IntegerAttr iota_dimension = in matchAndRewrite() local