Searched refs:ReplaceFunction (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 594 fld->ReplaceFunction(host_graph_func_name, host_graph_fdef)); in ConstructHostGraph()
799 fld->ReplaceFunction(shape_inference_graph_name, fdef_replace)); in RewriteShapeInferenceGraph()
869 TF_RETURN_IF_ERROR(fld->ReplaceFunction(func_name, replace_fdef)); in ReplaceKeyPlaceholderWithArgNode()
950 fld->ReplaceFunction(host_graph_func_name, oc_host_graph_fdef)); in BuildHostGraphForIfNode()
1015 TF_RETURN_IF_ERROR(fld->ReplaceFunction(loop_cond_func.name(), replace_fdef)); in AddSendLoopPredToLoopCond()
1095 fld->ReplaceFunction(cond_host_func_name, cond_replace_fdef)); in RewriteHostWhileLoopCond()
1157 fld->ReplaceFunction(body_host_func_name, body_replace_fdef)); in RewriteHostWhileLoopBody()
1220 fld->ReplaceFunction(host_graph_func_name, oc_host_graph_fdef)); in BuildHostGraphForWhileNode()
1268 fld->ReplaceFunction(host_graph_func_name, oc_host_graph_fdef)); in BuildHostGraphForFuncCallNode()
1660 TF_RETURN_IF_ERROR(fld->ReplaceFunction(shape_inference_graph.name(), in ExtractOutsideCompilationForFunction()
[all …]
|
| D | encapsulate_subgraphs_pass.cc | 1137 TF_RETURN_IF_ERROR(library->ReplaceFunction(name, fdef)); in BuildFunctionDef()
1199 TF_RETURN_IF_ERROR(library->ReplaceFunction(name, fdef)); in ReplaceFunctionDef()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 221 fld->ReplaceFunction(new_func_name, functionalized_fdef)); in FunctionalizeControlFlowForFunction()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | rebatch.cc | 228 TF_RETURN_IF_ERROR(flib->ReplaceFunction(func_name, optimized_func)); in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 643 TF_RETURN_IF_ERROR(flib.ReplaceFunction(func_name, optimized_func)); in Optimize()
766 TF_RETURN_IF_ERROR(flib->ReplaceFunction(func_name, fdef)); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 352 Status ReplaceFunction(const string& func, const FunctionDef& fdef);
|
| D | function.cc | 1201 Status FunctionLibraryDefinition::ReplaceFunction(const string& func, in ReplaceFunction() function in tensorflow::FunctionLibraryDefinition
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 707 TF_RETURN_IF_ERROR((*optimized_flib)->ReplaceFunction(func_name, fdef)); in OptimizeGraph()
|