| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| const_analysis.cc | 149 const FunctionBody* fbody = flib_runtime->GetFunctionBody(body_handle); local
151 TF_RET_CHECK(fbody);
152 int num_inputs = fbody->fdef.signature().input_arg_size();
160 *(fbody->graph), &compile_time_const_arg_indices,
173 Node* arg_i = fbody->arg_nodes[i];
174 Node* ret_i = fbody->ret_nodes[i];
|
| graph_compiler.cc | 244 const FunctionBody* fbody; local
245 TF_RETURN_IF_ERROR(compiler->FindFunctionBody(func, &fbody));
247 auto graph = compiler->GetGraph(fbody);
|
| tf2xla_util.cc | 187 FunctionBody* fbody; local
193 &fbody));
194 std::unique_ptr<FunctionBody> fbody_deleter(fbody);
197 Graph* func_graph = fbody->graph;
|
| xla_compiler.cc | 471 const FunctionBody** fbody) {
476 *fbody = flib_runtime->GetFunctionBody(handle);
477 TF_RET_CHECK(*fbody);
482 const FunctionBody** fbody) {
486 auto status = GetFunctionBody(function, local_flib_runtime_, fbody);
492 GetFunctionBody(function, flib_runtime_, fbody),
501 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) {
503 CopyGraph(*fbody->graph, graph.get());
557 const FunctionBody* fbody; local
558 TF_RETURN_IF_ERROR(FindFunctionBody(function, &fbody));
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 109 // runtime, returns this function's body in `fbody` as well as the indices
111 // `fbody` is owned by `flr`.
116 const FunctionBody** fbody,
124 *fbody = flr->GetFunctionBody(handle);
125 CHECK(*fbody); // Can't be nullptr since we just instantiated it.
126 const DataTypeVector& arg_types = (*fbody)->arg_types;
130 BackwardsConstAnalysis(*((*fbody)->graph), &const_args,
171 const FunctionBody* fbody = nullptr; local
175 flr, node_def, &fbody, &constant_arg_indices, &resource_arg_indices));
178 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY)
[all...] |
| extract_outside_compilation_pass.cc | 616 FunctionBody* fbody = nullptr; local
622 &fbody));
623 std::unique_ptr<FunctionBody> fbody_deleter(fbody);
624 Graph* host_graph = fbody->graph;
693 FunctionBody* fbody = nullptr; local
699 &fbody));
700 std::unique_ptr<FunctionBody> fbody_deleter(fbody);
701 Graph* g = fbody->graph;
833 FunctionBody* fbody = nullptr; local
839 &fbody));
964 FunctionBody* fbody = nullptr; local
1604 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
[all...] |
| mark_for_compilation_pass.cc | 240 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); local
241 CHECK(fbody);
242 const FunctionDef& fdef = fbody->fdef;
256 for (Node* node : fbody->graph->op_nodes()) {
[all...] |
| encapsulate_subgraphs_pass.cc | 2322 FunctionBody* fbody = nullptr; local
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| lower_if_op.cc | 243 FunctionBody* fbody; local
249 &fbody));
255 Status can_inline_function_call = ValidateInlining(n, fbody, inline_opts);
259 InlineFunctionBody(g->flib_def(), g, n, fbody, inline_opts));
265 delete fbody;
|
| lower_while_op.cc | 380 FunctionBody* fbody; local
386 &fbody));
392 Status can_inline_function_call = ValidateInlining(n, fbody, inline_opts);
396 InlineFunctionBody(g->flib_def(), g, n, fbody, inline_opts));
402 delete fbody;
|
| process_function_library_runtime.cc | 496 std::unique_ptr<FunctionBody> fbody(tmp_fbody);
497 *graph = std::unique_ptr<Graph>(fbody->graph);
498 fbody->graph = nullptr;
499 ret_node_names->reserve(fbody->ret_nodes.size());
500 for (const Node* node : fbody->ret_nodes) {
503 control_ret_node_names->reserve(fbody->control_ret_nodes.size());
504 for (const Node* node : fbody->control_ret_nodes) {
717 const FunctionBody* fbody = target_flr->GetFunctionBody(comp_data.handle_); local
718 DCHECK(fbody != nullptr);
723 if (fbody->ret_types[j] == DT_RESOURCE)
[all...] |
| function.cc | 405 FunctionBody** fbody);
571 const FunctionBody* fbody = GetFunctionBody(handle); local
572 CHECK_NOTNULL(fbody);
579 for (const auto& t : fbody->arg_types) {
583 for (const auto& t : fbody->ret_types) {
591 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types,
592 fbody->ret_types, output_memory_types, graph_def_version_, &s);
601 const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) {
603 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody);
702 FunctionBody* fbody = nullptr; local
859 const FunctionBody* fbody; local
970 const FunctionBody* fbody = GetFunctionBody(handle); local
1087 const FunctionBody* fbody = GetFunctionBody(handle); local
1814 const FunctionBody* fbody = lib->GetFunctionBody(handle); local
[all...] |
| function_test.cc | 300 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
301 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types);
336 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
337 CHECK_NOTNULL(fbody);
339 CopyGraph(*fbody->graph, ret.get());
352 const FunctionBody* fbody = flr->GetFunctionBody(handle); local
353 CHECK_NOTNULL(fbody);
354 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody));
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/ |
| captured_function.cc | 146 const FunctionBody* fbody = lib->GetFunctionBody(f_handle); local
147 if (fbody == nullptr) {
152 for (const auto& ret_type : fbody->ret_types) {
|
