| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 56 const FunctionBody* fbody = flr->GetFunctionBody(handle); in CreateXlaLaunchOp() local
57 CHECK(fbody); // Can't be nullptr since we just instantiated it. in CreateXlaLaunchOp()
58 std::vector<bool> const_args(fbody->arg_types.size()); in CreateXlaLaunchOp()
60 TF_RETURN_IF_ERROR(BackwardsConstAnalysis(*(fbody->graph), &const_args)); in CreateXlaLaunchOp()
66 DebugString(fbody->fdef)); in CreateXlaLaunchOp()
76 AddNodeAttr("Targs", fbody->arg_types, &launch_def); in CreateXlaLaunchOp()
77 AddNodeAttr("Tresults", fbody->ret_types, &launch_def); in CreateXlaLaunchOp()
86 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY); in CreateXlaLaunchOp()
87 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY); in CreateXlaLaunchOp()
94 &fbody->fdef.signature(), flr, fbody->arg_types, input_memory_types, in CreateXlaLaunchOp()
[all …]
|
| D | mark_for_compilation_pass.cc | 124 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
125 CHECK(fbody); in IsCompilableCall()
126 const FunctionDef& fdef = fbody->fdef; in IsCompilableCall()
139 for (Node* node : fbody->graph->op_nodes()) { in IsCompilableCall()
|
| D | encapsulate_subgraphs_pass.cc | 1848 FunctionBody* fbody = nullptr; in MakePrunedGraphCopyAndInline() local
1854 &fbody)); in MakePrunedGraphCopyAndInline()
1855 InlineFunctionBody(*library, pruned_graph->get(), node, fbody); in MakePrunedGraphCopyAndInline()
1856 delete fbody; in MakePrunedGraphCopyAndInline()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 230 FunctionBody** fbody);
384 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
385 CHECK_NOTNULL(fbody); in CreateKernel()
392 for (const auto& t : fbody->arg_types) { in CreateKernel()
397 for (const auto& t : fbody->ret_types) { in CreateKernel()
405 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types, in CreateKernel()
406 fbody->ret_types, output_memory_types, graph_def_version_, &s); in CreateKernel()
416 const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) { in FunctionDefToBody() argument
418 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody); in FunctionDefToBody()
423 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody); in FunctionDefToBody()
[all …]
|
| D | function.h | 160 Node* caller, const FunctionBody* fbody);
168 FunctionBody** fbody);
|
| D | function_test.cc | 276 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface() local
277 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
312 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody() local
313 CHECK_NOTNULL(fbody); in GetFuncBody()
315 CopyGraph(*fbody->graph, ret.get()); in GetFuncBody()
328 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody() local
329 CHECK_NOTNULL(fbody); in GetGradBody()
330 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody)); in GetGradBody()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 131 const FunctionBody** fbody) { in GetFunctionBody() argument
136 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
137 TF_RET_CHECK(*fbody); in GetFunctionBody()
142 const FunctionBody** fbody) { in FindFunctionBody() argument
146 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
152 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
158 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph() argument
160 CopyGraph(*fbody->graph, graph.get()); in GetGraph()
187 const FunctionBody* fbody; in CompileFunction() local
188 TF_RETURN_IF_ERROR(FindFunctionBody(function, &fbody)); in CompileFunction()
[all …]
|
| D | graph_compiler.cc | 200 const FunctionBody* fbody; in CompileFunctionalNode() local
201 TF_RETURN_IF_ERROR(compiler->FindFunctionBody(func, &fbody)); in CompileFunctionalNode()
203 auto graph = compiler->GetGraph(fbody); in CompileFunctionalNode()
|
| D | xla_compiler.h | 306 const FunctionBody** fbody);
309 std::unique_ptr<Graph> GetGraph(const FunctionBody* fbody);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 173 const FunctionBody* fbody = lib_->GetFunctionBody(f_handle_); in MaybeInstantiate() local
174 if (fbody == nullptr) { in MaybeInstantiate()
177 ret_types_ = fbody->ret_types; in MaybeInstantiate()
|
| /external/mksh/src/ |
| D | check.t | 11289 fbody=$fbody$'\n'$line
11292 fbody=$fbody$'\n\t'$line
11296 x="inline_${nextf}() {"$fbody$'\n}\n'
11299 x="function comsub_$nextf { x=\$("$fbody$'\n); }\n'
11302 x="function reread_$nextf { x=\$(("$fbody$'\n)|tr u x); }\n'
11307 fbody=
11990 fbody=$fbody$'\n'$line
11993 fbody=$fbody$'\n\t'$line
11997 x="inline_${nextf}() {"$fbody$'\n}\n'
12000 x="function comsub_$nextf { x=\$("$fbody$'\n); }\n'
[all …]
|