Searched refs:defining_op (Results 1 – 11 of 11) sorted by relevance
34 auto defining_op = op->getOperand(0).getDefiningOp(); in matchAndRewrite() local36 llvm::dyn_cast_or_null<InferShapedTypeOpInterface>(defining_op); in matchAndRewrite()40 op->getContext(), op->getLoc(), defining_op->getOperands(), in matchAndRewrite()41 defining_op->getAttrDictionary(), defining_op->getRegions(), in matchAndRewrite()72 auto defining_op = llvm::dyn_cast_or_null<InferShapedTypeOpInterface>( in matchAndRewrite() local74 if (!defining_op) return failure(); in matchAndRewrite()76 if (failed(defining_op.reifyReturnTypeShapes(rewriter, return_shapes))) { in matchAndRewrite()
80 Operation* defining_op = operand.getDefiningOp(); in CanMergeIntoCluster() local83 if (defining_op->getBlock() != c.ops.front()->getBlock()) return true; in CanMergeIntoCluster()86 if (defining_op->isBeforeInBlock(c.ops.front())) return true; in CanMergeIntoCluster()93 if (defining_op->isBeforeInBlock(c.ops.back()->getNextNode()) && in CanMergeIntoCluster()94 GetDevice(defining_op) == c.device) in CanMergeIntoCluster()
106 Operation *defining_op = arg.getDefiningOp(); in runOnFunction() local108 defining_op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction()
205 if (auto* defining_op = v.getDefiningOp()) { in GetExternalOperands() local206 if (!op->isAncestor(defining_op) && in GetExternalOperands()207 tpu_cluster->isAncestor(defining_op) && in GetExternalOperands()208 !HasOutsideCompilationAncestor(defining_op) && in GetExternalOperands()209 !llvm::isa<TF::_XlaRecvAtHostV2Op>(defining_op)) { in GetExternalOperands()
89 Operation* defining_op = operand.getDefiningOp(); in SortTopologically() local90 if (!defining_op) continue; in SortTopologically()92 block->findAncestorOpInBlock(*defining_op); in SortTopologically()
144 Operation *defining_op = value.getDefiningOp(); in GetFunctionMetadatas() local145 std::string defining_op_host = GetHost(defining_op); in GetFunctionMetadatas()149 defining_op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetFunctionMetadatas()
195 auto defining_op = operand.getDefiningOp(); in FindSourcesAndSinksInIsland() local196 if (!defining_op || defining_op->getParentOp() != island) continue; in FindSourcesAndSinksInIsland()198 result.sinks.erase(defining_op); in FindSourcesAndSinksInIsland()
441 Operation* defining_op = in GetControlRetNodes() local443 auto node_it = nodes_.find(defining_op); in GetControlRetNodes()
265 auto defining_op = terminator->getOperand(i).getDefiningOp(); in convert_graph_uint8_tensor() local268 if (!defining_op) continue; in convert_graph_uint8_tensor()269 Value input_val = defining_op->getResult(0); in convert_graph_uint8_tensor()
543 auto defining_op = operand.getDefiningOp(); in GetExprForOperand() local544 if (defining_op && matchPattern(defining_op, m_Constant(&const_attr))) { in GetExprForOperand()
817 absl::flat_hash_map<int32_t, const tflite::OperatorT*> defining_op; in PruneSubgraph() local821 defining_op[output] = op.get(); in PruneSubgraph()828 if (auto& op = defining_op[output]) { in PruneSubgraph()846 if (auto& op = defining_op[input]) { in PruneSubgraph()