Searched refs:operand_op (Results 1 – 5 of 5) sorted by relevance
77 Operation* operand_op = GetOpOfValue(operand); in ExpandHeadOutsideCompiledOps() local78 if (head_outside_compiled_ops.count(operand_op)) continue; in ExpandHeadOutsideCompiledOps()80 if (operand_op->getParentRegion() == cluster_region) in ExpandHeadOutsideCompiledOps()
166 Operation* operand_op = GetOpOfValue(operand); in FindOutsideCompiledOpsAtHead() local167 if (head_outside_compiled_ops.count(operand_op) || in FindOutsideCompiledOpsAtHead()168 operand_op == &cluster_op) in FindOutsideCompiledOpsAtHead()171 if (operand_op->getParentRegion() == cluster_region) in FindOutsideCompiledOpsAtHead()276 Operation* operand_op = GetOpOfValue(operand); in FindOutsideCompiledOpsAtTailAndClusterResults() local277 if (operand_op->getParentRegion() == cluster_region) in FindOutsideCompiledOpsAtTailAndClusterResults()
256 if (Operation* operand_op = operand.get().getDefiningOp()) in MoveTransposeBefore() local257 work_list->push_back(operand_op); in MoveTransposeBefore()
154 auto operand_op = operand.getDefiningOp(); in AllOperationSafe() local155 if (IsConstant(operand_op)) continue; in AllOperationSafe()156 if (operand_op->getBlock() != op->getBlock()) { in AllOperationSafe()
533 TFOp operand_op = operand.getDefiningOp(); in IsFoldableUncached() local534 if (operand_op && dialect_->IsConstant(operand_op)) { in IsFoldableUncached()535 auto dtype = operand_op->getAttrOfType<TypeAttr>("dtype"); in IsFoldableUncached()540 merge_has_constant_input |= operand_op.getControlOperands().empty(); in IsFoldableUncached()1319 Operation *operand_op = operand.getDefiningOp(); in matchAndRewrite() local1320 if (!operand_op) continue; in matchAndRewrite()1321 if (!this->dialect_->IsConstant(operand_op)) continue; in matchAndRewrite()1322 if (!TFOp(operand_op).getControlOperands().empty()) continue; in matchAndRewrite()1326 *(operand_op->result_type_begin()), TFOp(op).controlRet(), in matchAndRewrite()1327 operand_op->getAttrOfType<ElementsAttr>("value"), op->getAttrs()); in matchAndRewrite()