| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 56 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
57 CHECK(fbody); // Can't be nullptr since we just instantiated it.
58 std::vector<bool> const_args(fbody->arg_types.size());
60 TF_RETURN_IF_ERROR(BackwardsConstAnalysis(*(fbody->graph), &const_args));
66 DebugString(fbody->fdef));
76 AddNodeAttr("Targs", fbody->arg_types, &launch_def);
77 AddNodeAttr("Tresults", fbody->ret_types, &launch_def);
86 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY);
87 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY);
94 &fbody->fdef.signature(), flr, fbody->arg_types, input_memory_types
[all...] |
| mark_for_compilation_pass.cc | 124 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); local
125 CHECK(fbody);
126 const FunctionDef& fdef = fbody->fdef;
139 for (Node* node : fbody->graph->op_nodes()) {
|
| encapsulate_subgraphs_pass.cc | 1848 FunctionBody* fbody = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| captured_function.cc | 173 const FunctionBody* fbody = lib_->GetFunctionBody(f_handle_); local
174 if (fbody == nullptr) {
177 ret_types_ = fbody->ret_types;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| graph_compiler.cc | 200 const FunctionBody* fbody; local
201 TF_RETURN_IF_ERROR(compiler->FindFunctionBody(func, &fbody));
203 auto graph = compiler->GetGraph(fbody);
|
| xla_compiler.cc | 131 const FunctionBody** fbody) {
136 *fbody = flib_runtime->GetFunctionBody(handle);
137 TF_RET_CHECK(*fbody);
142 const FunctionBody** fbody) {
146 auto status = GetFunctionBody(function, local_flib_runtime_, fbody);
152 GetFunctionBody(function, flib_runtime_, fbody),
158 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) {
160 CopyGraph(*fbody->graph, graph.get());
187 const FunctionBody* fbody; local
188 TF_RETURN_IF_ERROR(FindFunctionBody(function, &fbody));
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 230 FunctionBody** fbody);
384 const FunctionBody* fbody = GetFunctionBody(handle); local
385 CHECK_NOTNULL(fbody);
392 for (const auto& t : fbody->arg_types) {
397 for (const auto& t : fbody->ret_types) {
405 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types,
406 fbody->ret_types, output_memory_types, graph_def_version_, &s);
416 const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) {
418 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody);
494 FunctionBody* fbody = nullptr; local
598 const FunctionBody* fbody; local
692 const FunctionBody* fbody = GetFunctionBody(handle); local
798 const FunctionBody* fbody = GetFunctionBody(handle); local
1330 const FunctionBody* fbody = lib->GetFunctionBody(handle); local
[all...] |
| function_test.cc | 276 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
277 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types);
312 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
313 CHECK_NOTNULL(fbody);
315 CopyGraph(*fbody->graph, ret.get());
328 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
329 CHECK_NOTNULL(fbody);
330 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody));
[all...] |
