Searched refs:iota_dimension_ (Results 1 – 2 of 2) sorted by relevance
1441 int64 iota_dimension() const { return iota_dimension_; } in iota_dimension()1457 const int64 iota_dimension_; variable
2482 iota_dimension_(iota_dimension) {} in HloIotaInstruction()