| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops_pass.cc | 95 auto walk_result = in ApplyPatternsLocallyUntilConverged() local
103 if (walk_result.wasInterrupted()) return failure(); in ApplyPatternsLocallyUntilConverged()
136 auto walk_result = func.walk([&](tf_device::ClusterOp cluster) { in ApplyPatternsInClusterAndReachableFunctions() local
147 if (walk_result.wasInterrupted()) return failure(); in ApplyPatternsInClusterAndReachableFunctions()
|
| D | executor_tpuv1_inline_tpu_island.cc | 57 auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { in runOnOperation() local
79 if (walk_result.wasInterrupted()) return signalPassFailure(); in runOnOperation()
|
| D | tpu_host_computation_expansion.cc | 75 auto walk_result = cluster_op.walk([&](Operation* op) { in ExpandHeadOutsideCompiledOps() local
86 if (!walk_result.wasInterrupted()) in ExpandHeadOutsideCompiledOps()
|
| D | tpu_update_embedding_enqueue_op_inputs.cc | 69 auto walk_result = func_op.walk([&](Operation* op) { in FindTPUEmbeddingOps() local
86 return failure(walk_result.wasInterrupted()); in FindTPUEmbeddingOps()
|
| D | tpu_extract_head_tail_outside_compilation.cc | 164 auto walk_result = cluster_op.walk([&](Operation* op) { in FindOutsideCompiledOpsAtHead() local
177 if (!walk_result.wasInterrupted()) in FindOutsideCompiledOpsAtHead()
448 auto walk_result = module.walk([&](tf_device::ClusterOp cluster) { in CheckPostconditions() local
452 if (walk_result.wasInterrupted()) return failure(); in CheckPostconditions()
|
| D | resource_op_lifting_cleanup.cc | 408 auto walk_result = parent_op->walk([](Operation *op) { in CleanupAndCanonicalize() local
451 return failure(walk_result.wasInterrupted()); in CleanupAndCanonicalize()
463 auto walk_result = module.walk([](tf_device::ClusterOp cluster) { in CleanupAndCanonicalizeForResourceOpLifting() local
467 return failure(walk_result.wasInterrupted()); in CleanupAndCanonicalizeForResourceOpLifting()
|
| D | resource_op_lifting.cc | 1247 auto walk_result = module.walk([&](func::FuncOp func_op) { in runOnOperation() local
1259 if (walk_result.wasInterrupted()) return signalPassFailure(); in runOnOperation()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | designate_resource_handle_mesh.cc | 72 auto walk_result = in runOnOperation() local
80 if (walk_result.wasInterrupted()) return signalPassFailure(); in runOnOperation()
|
| D | op_to_device_cluster.cc | 115 auto walk_result = getOperation().walk([&](mlir::Operation *operation) { in runOnOperation() local
133 if (walk_result.wasInterrupted()) signalPassFailure(); in runOnOperation()
|
| D | tpu_add_resource_device_attribute.cc | 95 mlir::WalkResult walk_result = in runOnOperation() local
127 if (walk_result.wasInterrupted()) return signalPassFailure(); in runOnOperation()
|
| D | dtensor_mixed_precision_reduce.cc | 123 mlir::WalkResult walk_result = function.walk([&](ReduceOpType reduce_op) { in TryMixedPrecisionReduce() local
133 if (walk_result.wasInterrupted()) return mlir::failure(); in TryMixedPrecisionReduce()
|
| D | propagate_default_layout.cc | 167 auto walk_result = in runOnOperation() local
208 if (walk_result.wasInterrupted()) return signalPassFailure(); in runOnOperation()
|
| D | merge_clusters.cc | 463 mlir::WalkResult walk_result = cluster->walk([&](mlir::Operation* op) { in DecomposeControlflow() local
471 if (walk_result.wasInterrupted()) return mlir::failure(); in DecomposeControlflow()
|
| D | mesh_propagation.cc | 521 auto walk_result = function.walk([&](mlir::tf_device::ClusterOp cluster) { in PropagateDefaultMeshToUnAssignedClusters() local
536 if (walk_result.wasInterrupted()) return mlir::failure(); in PropagateDefaultMeshToUnAssignedClusters()
|
| D | layout_propagation_v2.cc | 322 auto walk_result = module.walk([&](mlir::Operation* op) { in InsertInitialLayoutsFromComputeLayout() local
393 return mlir::failure(walk_result.wasInterrupted()); in InsertInitialLayoutsFromComputeLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | trim_functions_tf.cc | 104 auto walk_result = func.walk([&](func::CallOp op) -> WalkResult { in Verify() local
110 if (walk_result.wasInterrupted()) return signalPassFailure(); in Verify()
|
| D | reduce_while_operands.cc | 142 auto walk_result = block.walk([&](Operation *op) { in AllOperationSafe() local
162 return !walk_result.wasInterrupted(); in AllOperationSafe()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.cc | 58 auto walk_result = module.walk([&](mlir::Operation* op) { in HasTPUOp() local
83 return walk_result.wasInterrupted(); in HasTPUOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | decompose.cc | 290 auto walk_result = func.walk([&](CallOp call_op) { in InlineTFRFuncCalls() local
323 if (walk_result.wasInterrupted()) { in InlineTFRFuncCalls()
|