Home
last modified time | relevance | path

Searched refs:operand_constraints_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h217 std::map<OperandConstraintKey, OperandLayoutConstraint> operand_constraints_; variable
Dlayout_assignment.cc245 auto iter = operand_constraints_.find(key); in SetOperandLayout()
246 if (iter == operand_constraints_.end()) { in SetOperandLayout()
250 iter = operand_constraints_.insert(pair).first; in SetOperandLayout()
356 auto it = operand_constraints_.find(std::make_pair(instruction, operand_no)); in GetOperandLayoutConstraint()
357 return it == operand_constraints_.end() ? nullptr : &it->second; in GetOperandLayoutConstraint()