Searched refs:cond_arg (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | control_flow_spmd_expander.cc | 39 mlir::BlockArgument cond_arg = std::get<0>(data.value()); in ExpandOp() local
41 cond_arg.setType(while_op.getOperand(index).getType()); in ExpandOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | prepare_for_export.cc | 99 Value cond_arg = in prepareWhileOp() local
105 operand.set(cond_arg); in prepareWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | resource_op_lifting_cleanup.cc | 207 BlockArgument cond_arg = cond.getArgument(result_idx); in EliminateUnusedResultsForWhile() local
212 if (cond_arg.use_empty() && body_arg.hasOneUse() && in EliminateUnusedResultsForWhile()
388 Value cond_arg = cond.getArgument(result_idx); in CanonicalizeWhileRegion() local
394 cond_arg.replaceAllUsesWith(while_operand); in CanonicalizeWhileRegion()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3208 auto cond_arg = cond_block.getArgument(op_idx); in matchAndRewrite() local
3209 if (cond_arg.getType() == while_operand.getType()) in matchAndRewrite()
3210 cond_arg.replaceAllUsesWith(while_operand); in matchAndRewrite()
|
| /external/webrtc/third_party/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 504 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
509 cond(cond_arg), in SynchWaitParams()
|
| /external/cronet/third_party/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 507 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
512 cond(cond_arg), in SynchWaitParams()
|
| /external/libtextclassifier/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 468 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
473 cond(cond_arg), in SynchWaitParams()
|
| /external/openscreen/third_party/abseil/src/absl/synchronization/ |
| D | mutex.cc | 468 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
473 cond(cond_arg), in SynchWaitParams()
|
| /external/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 479 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
484 cond(cond_arg), in SynchWaitParams()
|
| /external/tensorflow/third_party/absl/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 486 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
491 cond(cond_arg), in SynchWaitParams()
|
| /external/angle/third_party/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 496 SynchWaitParams(Mutex::MuHow how_arg, const Condition* cond_arg, in SynchWaitParams()
501 cond(cond_arg), in SynchWaitParams()
|
| /external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/synchronization/ |
| D | mutex.cc | 479 SynchWaitParams(Mutex::MuHow how_arg, const Condition *cond_arg, in SynchWaitParams()
484 cond(cond_arg), in SynchWaitParams()
|