Searched defs:iota_dimension (Results 1 – 6 of 6) sorted by relevance
1441 int64 iota_dimension() const { return iota_dimension_; } in iota_dimension() function
408 int64 iota_dimension = iota->iota_dimension(); in HandleIota() local
719 optional<int64> iota_dimension; in ParseInstructionRhs() local
654 const Shape& shape, int64 iota_dimension) { in CreateIota()
2480 HloIotaInstruction::HloIotaInstruction(const Shape& shape, int64 iota_dimension) in HloIotaInstruction()
598 XlaOp XlaBuilder::Iota(const Shape& shape, int64 iota_dimension) { in Iota()3513 XlaOp Iota(XlaBuilder* builder, const Shape& shape, int64 iota_dimension) { in Iota()