Searched refs:WrapsSingleOp (Results 1 – 15 of 15) sorted by relevance
127 if (!island || !island.WrapsSingleOp()) return success(); in MergeIsland()147 if (!candidate_island || !candidate_island.WrapsSingleOp()) continue; in MergeIsland()164 if (!wrapper || !wrapper.WrapsSingleOp()) continue; in MergeIsland()
287 if (!compile_launch || !compile_launch.WrapsSingleOp()) return; in runOnFunction()298 if (!execute_launch || !execute_launch.WrapsSingleOp()) return; in runOnFunction()
111 if (op.WrapsSingleOp() && in runOnFunction()
187 if (!island_op.WrapsSingleOp()) continue; in runOnFunction()
82 if (!island_op || island_op.WrapsSingleOp()) return; in runOnOperation()
389 if (!execute_launch_op.WrapsSingleOp() || in HandleReplicateOp()409 if (!compile_launch || !compile_launch.WrapsSingleOp() || in HandleReplicateOp()
170 if (!island_op.WrapsSingleOp()) continue; in PropagateDevicesInGraph()
311 if (!island_op.WrapsSingleOp()) continue; in runOnFunction()
601 if (op.WrapsSingleOp() && in runOnFunction()
128 bool LaunchOp::WrapsSingleOp() { return BlockWrapsSingleOp(&GetBody()); } in WrapsSingleOp() function in mlir::tf_device::LaunchOp670 bool ReplicateOp::WrapsSingleOp() { return BlockWrapsSingleOp(&GetBody()); } in WrapsSingleOp() function in mlir::tf_device::ReplicateOp
75 bool WrapsSingleOp();293 bool WrapsSingleOp();
266 bool IslandOp::WrapsSingleOp() { in WrapsSingleOp() function in mlir::tf_executor::IslandOp324 if (op.getAttrs().empty() && op.WrapsSingleOp()) { in Print()
212 bool WrapsSingleOp();
223 if (island_op.WrapsSingleOp()) return; in BreakUpIsland()
126 if (!island.WrapsSingleOp()) { in HasSingleGraphSingleOpIslandsFunctions()