| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops_pass.cc | 95 auto walk_result = in ApplyPatternsLocallyUntilConverged() local
136 auto walk_result = func.walk([&](tf_device::ClusterOp cluster) { in ApplyPatternsInClusterAndReachableFunctions() local
|
| D | executor_tpuv1_inline_tpu_island.cc | 57 auto walk_result = getOperation().walk([&](TF::PartitionedCallOp call_op) { in runOnOperation() local
|
| D | tpu_host_computation_expansion.cc | 75 auto walk_result = cluster_op.walk([&](Operation* op) { in ExpandHeadOutsideCompiledOps() local
|
| D | tpu_update_embedding_enqueue_op_inputs.cc | 69 auto walk_result = func_op.walk([&](Operation* op) { in FindTPUEmbeddingOps() local
|
| D | resource_op_lifting_cleanup.cc | 408 auto walk_result = parent_op->walk([](Operation *op) { in CleanupAndCanonicalize() local
463 auto walk_result = module.walk([](tf_device::ClusterOp cluster) { in CleanupAndCanonicalizeForResourceOpLifting() local
|
| D | tpu_extract_head_tail_outside_compilation.cc | 164 auto walk_result = cluster_op.walk([&](Operation* op) { in FindOutsideCompiledOpsAtHead() local
448 auto walk_result = module.walk([&](tf_device::ClusterOp cluster) { in CheckPostconditions() local
|
| D | resource_op_lifting.cc | 1247 auto walk_result = module.walk([&](func::FuncOp func_op) { in runOnOperation() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | designate_resource_handle_mesh.cc | 72 auto walk_result = in runOnOperation() local
|
| D | op_to_device_cluster.cc | 115 auto walk_result = getOperation().walk([&](mlir::Operation *operation) { in runOnOperation() local
|
| D | dtensor_mixed_precision_reduce.cc | 123 mlir::WalkResult walk_result = function.walk([&](ReduceOpType reduce_op) { in TryMixedPrecisionReduce() local
|
| D | tpu_add_resource_device_attribute.cc | 95 mlir::WalkResult walk_result = in runOnOperation() local
|
| D | propagate_default_layout.cc | 167 auto walk_result = in runOnOperation() local
|
| D | merge_clusters.cc | 463 mlir::WalkResult walk_result = cluster->walk([&](mlir::Operation* op) { in DecomposeControlflow() local
|
| D | mesh_propagation.cc | 521 auto walk_result = function.walk([&](mlir::tf_device::ClusterOp cluster) { in PropagateDefaultMeshToUnAssignedClusters() local
|
| D | layout_propagation_v2.cc | 322 auto walk_result = module.walk([&](mlir::Operation* op) { in InsertInitialLayoutsFromComputeLayout() local
|
| /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
|
| D | reduce_while_operands.cc | 142 auto walk_result = block.walk([&](Operation *op) { in AllOperationSafe() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.cc | 58 auto walk_result = module.walk([&](mlir::Operation* op) { in HasTPUOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | decompose.cc | 290 auto walk_result = func.walk([&](CallOp call_op) { in InlineTFRFuncCalls() local
|