Searched refs:indexed_generators_ (Results 1 – 2 of 2) sorted by relevance
44 indexed_generators_[hlo] = in DefaultAction()75 hlo, indexed_generators_)(index)); in DefaultAction()85 indexed_generators_[constant] = [=](const IrArray::Index& index) { in HandleConstant()117 if (indexed_generators_.find(parameter) == indexed_generators_.end()) { in HandleParameter()130 indexed_generators_[tuple] = in HandleTuple()136 indexed_generators_[operands[i]](index)); in HandleTuple()167 if (indexed_generators_.count(instr)) { in GetGenerator()190 CHECK(indexed_generators_.count(instr)); in GetGenerator()192 return indexed_generators_.at(instruction); in GetGenerator()
64 indexed_generators_[hlo] = std::move(generator); in BindGenerator()100 indexed_generators_; variable