Searched defs:ClusterOp (Results 1 – 2 of 2) sorted by relevance
45 llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter) in ClusterConstantSinkingPass()99 llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter) { in CreateClusterConstantSinkingPass()
26 std::function<WalkResult(Operation*, tf_device::ClusterOp, in WalkReachableFromTpuCluster()98 ModuleOp module, std::function<WalkResult(Operation*, tf_device::ClusterOp, in WalkReachableFromTpuCluster()