Searched refs:RewriteFunctionCallNode (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_function_call_op.h | 31 Status RewriteFunctionCallNode(Node* n, Graph* g,
|
| H A D | lower_function_call_op.cc | 32 Status RewriteFunctionCallNode(Node* n, Graph* g, in RewriteFunctionCallNode() function
|
| H A D | lower_functional_ops.cc | 172 TF_RETURN_IF_ERROR(RewriteFunctionCallNode(n, g, *flib_def, in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | host_compute_ops.cc | 291 TF_RETURN_IF_ERROR(RewriteFunctionCallNode( in LowerFunctionalOps()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 4247 RewriteFunctionCallNode(n, g, flib_def, in LowerOutsideCompilationFunctionalNodes()
|