| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 116 const FunctionBody** fbody, in GetBodyAndConstantsAndResources() argument
124 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
125 CHECK(*fbody); // Can't be nullptr since we just instantiated it. in GetBodyAndConstantsAndResources()
126 const DataTypeVector& arg_types = (*fbody)->arg_types; in GetBodyAndConstantsAndResources()
130 BackwardsConstAnalysis(*((*fbody)->graph), &const_args, in GetBodyAndConstantsAndResources()
171 const FunctionBody* fbody = nullptr; in CreateXlaLaunchOp() local
175 flr, node_def, &fbody, &constant_arg_indices, &resource_arg_indices)); in CreateXlaLaunchOp()
178 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY); in CreateXlaLaunchOp()
188 for (int i = 0; i < fbody->arg_types.size(); ++i) { in CreateXlaLaunchOp()
217 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY); in CreateXlaLaunchOp()
[all …]
|
| D | extract_outside_compilation_pass.cc | 616 FunctionBody* fbody = nullptr; in ExpandHostGraphIntoMainGraph() local
622 &fbody)); in ExpandHostGraphIntoMainGraph()
623 std::unique_ptr<FunctionBody> fbody_deleter(fbody); in ExpandHostGraphIntoMainGraph()
624 Graph* host_graph = fbody->graph; in ExpandHostGraphIntoMainGraph()
693 FunctionBody* fbody = nullptr; in RewriteShapeInferenceGraph() local
699 &fbody)); in RewriteShapeInferenceGraph()
700 std::unique_ptr<FunctionBody> fbody_deleter(fbody); in RewriteShapeInferenceGraph()
701 Graph* g = fbody->graph; in RewriteShapeInferenceGraph()
833 FunctionBody* fbody = nullptr; in ReplaceKeyPlaceholderWithArgNode() local
839 &fbody)); in ReplaceKeyPlaceholderWithArgNode()
[all …]
|
| D | mark_for_compilation_pass.cc | 240 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
241 CHECK(fbody); in IsCompilableCall()
242 const FunctionDef& fdef = fbody->fdef; in IsCompilableCall()
256 for (Node* node : fbody->graph->op_nodes()) { in IsCompilableCall()
|
| D | encapsulate_subgraphs_pass.cc | 2322 FunctionBody* fbody = nullptr; in MakePrunedGraphCopyAndInline() local
2328 &fbody)); in MakePrunedGraphCopyAndInline()
2334 fbody, inline_opts)); in MakePrunedGraphCopyAndInline()
2335 delete fbody; in MakePrunedGraphCopyAndInline()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 405 FunctionBody** fbody);
571 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
572 CHECK_NOTNULL(fbody); in CreateKernel()
579 for (const auto& t : fbody->arg_types) { in CreateKernel()
583 for (const auto& t : fbody->ret_types) { in CreateKernel()
591 &fbody->fdef.signature(), this, fbody->arg_types, input_memory_types, in CreateKernel()
592 fbody->ret_types, output_memory_types, graph_def_version_, &s); in CreateKernel()
601 const FunctionLibraryDefinition* lib_def, FunctionBody** fbody) { in FunctionDefToBody() argument
603 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig_, fbody); in FunctionDefToBody()
608 return FunctionDefToBodyHelper(fdef, attrs, lib_def, get_func_sig, fbody); in FunctionDefToBody()
[all …]
|
| D | lower_if_op.cc | 243 FunctionBody* fbody; in InlineCallInGraph() local
249 &fbody)); in InlineCallInGraph()
255 Status can_inline_function_call = ValidateInlining(n, fbody, inline_opts); in InlineCallInGraph()
259 InlineFunctionBody(g->flib_def(), g, n, fbody, inline_opts)); in InlineCallInGraph()
265 delete fbody; in InlineCallInGraph()
|
| D | function.h | 195 Status ValidateInlining(const Node* node, const FunctionBody* fbody,
207 Node* caller, const FunctionBody* fbody,
290 FunctionBody** fbody);
|
| D | lower_while_op.cc | 380 FunctionBody* fbody; in InlineCallInGraph() local
386 &fbody)); in InlineCallInGraph()
392 Status can_inline_function_call = ValidateInlining(n, fbody, inline_opts); in InlineCallInGraph()
396 InlineFunctionBody(g->flib_def(), g, n, fbody, inline_opts)); in InlineCallInGraph()
402 delete fbody; in InlineCallInGraph()
|
| D | process_function_library_runtime.cc | 496 std::unique_ptr<FunctionBody> fbody(tmp_fbody); in GetGraphAndRets() local
497 *graph = std::unique_ptr<Graph>(fbody->graph); in GetGraphAndRets()
498 fbody->graph = nullptr; in GetGraphAndRets()
499 ret_node_names->reserve(fbody->ret_nodes.size()); in GetGraphAndRets()
500 for (const Node* node : fbody->ret_nodes) { in GetGraphAndRets()
503 control_ret_node_names->reserve(fbody->control_ret_nodes.size()); in GetGraphAndRets()
504 for (const Node* node : fbody->control_ret_nodes) { in GetGraphAndRets()
717 const FunctionBody* fbody = target_flr->GetFunctionBody(comp_data.handle_); in GetOutputDevices() local
718 DCHECK(fbody != nullptr); in GetOutputDevices()
723 if (fbody->ret_types[j] == DT_RESOURCE) { in GetOutputDevices()
|
| D | function_test.cc | 300 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface() local
301 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
336 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody() local
337 CHECK_NOTNULL(fbody); in GetFuncBody()
339 CopyGraph(*fbody->graph, ret.get()); in GetFuncBody()
352 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody() local
353 CHECK_NOTNULL(fbody); in GetGradBody()
354 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody)); in GetGradBody()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 149 const FunctionBody* fbody = flib_runtime->GetFunctionBody(body_handle); in GetCompileTimeConstInputs() local
151 TF_RET_CHECK(fbody); in GetCompileTimeConstInputs()
152 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs()
160 *(fbody->graph), &compile_time_const_arg_indices, in GetCompileTimeConstInputs()
173 Node* arg_i = fbody->arg_nodes[i]; in GetCompileTimeConstInputs()
174 Node* ret_i = fbody->ret_nodes[i]; in GetCompileTimeConstInputs()
|
| D | xla_compiler.cc | 471 const FunctionBody** fbody) { in GetFunctionBody() argument
476 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
477 TF_RET_CHECK(*fbody); in GetFunctionBody()
482 const FunctionBody** fbody) { in FindFunctionBody() argument
486 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
492 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
501 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph() argument
503 CopyGraph(*fbody->graph, graph.get()); in GetGraph()
557 const FunctionBody* fbody; in CompileFunction() local
558 TF_RETURN_IF_ERROR(FindFunctionBody(function, &fbody)); in CompileFunction()
[all …]
|
| D | graph_compiler.cc | 244 const FunctionBody* fbody; in CompileFunctionalNode() local
245 TF_RETURN_IF_ERROR(compiler->FindFunctionBody(func, &fbody)); in CompileFunctionalNode()
247 auto graph = compiler->GetGraph(fbody); in CompileFunctionalNode()
|
| D | xla_compiler.h | 422 const FunctionBody** fbody);
426 std::unique_ptr<Graph> GetGraph(const FunctionBody* fbody);
|
| D | tf2xla_util.cc | 187 FunctionBody* fbody; in PropagateConstIntoFuncAttr() local
193 &fbody)); in PropagateConstIntoFuncAttr()
194 std::unique_ptr<FunctionBody> fbody_deleter(fbody); in PropagateConstIntoFuncAttr()
197 Graph* func_graph = fbody->graph; in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 146 const FunctionBody* fbody = lib->GetFunctionBody(f_handle); in Instantiate() local
147 if (fbody == nullptr) { in Instantiate()
152 for (const auto& ret_type : fbody->ret_types) { in Instantiate()
|
| /external/mksh/src/ |
| D | check.t | 11482 fbody=$fbody$'\n'$line
11485 fbody=$fbody$'\n\t'$line
11489 x="inline_${nextf}() {"$fbody$'\n}\n'
11492 x="function comsub_$nextf { x=\$("$fbody$'\n); }\n'
11495 x="function reread_$nextf { x=\$(("$fbody$'\n)|tr u x); }\n'
11500 fbody=
12183 fbody=$fbody$'\n'$line
12186 fbody=$fbody$'\n\t'$line
12190 x="inline_${nextf}() {"$fbody$'\n}\n'
12193 x="function comsub_$nextf { x=\$("$fbody$'\n); }\n'
[all …]
|