Searched refs:RewriteIfNode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.h | 27 Status RewriteIfNode(Node* n, Graph* g, bool keep_node_fetchable);
|
| D | lower_functional_ops.cc | 160 TF_RETURN_IF_ERROR(RewriteIfNode(n, g, keep_lowered_nodes_fetchable)); in Run()
|
| D | lower_if_op.cc | 300 Status RewriteIfNode(Node* n, Graph* g, bool keep_node_fetchable) { in RewriteIfNode() function
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 261 RewriteIfNode(if_node, g, /*keep_node_fetchable=*/false)); in LowerFunctionalOps()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1255 TF_RETURN_IF_ERROR(RewriteIfNode(n, graph.get(), false)); in InlineFunctionCalls()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3778 TF_RETURN_IF_ERROR(RewriteIfNode(n, g, /*keep_node_fetchable=*/false)); in LowerOutsideCompilationFunctionalNodes()
|