| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_util.cc | 87 const FunctionBody** fbody, in GetBodyAndConstantsAndResources() argument
98 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
99 CHECK(*fbody); // Can't be nullptr since we just instantiated it. in GetBodyAndConstantsAndResources()
100 const DataTypeVector& arg_types = (*fbody)->arg_types; in GetBodyAndConstantsAndResources()
104 BackwardsConstAnalysis(*((*fbody)->graph), &const_args, in GetBodyAndConstantsAndResources()
164 const FunctionBody* fbody = nullptr; in CreateXlaKernel() local
168 flr, node_def, &fbody, &constant_arg_indices, &resource_arg_indices)); in CreateXlaKernel()
171 MemoryTypeVector input_memory_types(fbody->arg_types.size(), DEVICE_MEMORY); in CreateXlaKernel()
181 for (int i = 0; i < fbody->arg_types.size(); ++i) { in CreateXlaKernel()
210 MemoryTypeVector output_memory_types(fbody->ret_types.size(), DEVICE_MEMORY); in CreateXlaKernel()
[all …]
|
| D | extract_outside_compilation_pass.cc | 807 std::unique_ptr<FunctionBody> fbody; in PostprocessLiftedArgsForCall() local
808 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fdef, n->attrs(), fld, &fbody)); in PostprocessLiftedArgsForCall()
810 int original_arg_count = fbody->arg_nodes.size(); in PostprocessLiftedArgsForCall()
814 *fbody, outside_compilation_attr_to_node)); in PostprocessLiftedArgsForCall()
843 AddOutsideCompilationInputArgToFunctionBody(*fbody, i, data_types[i])); in PostprocessLiftedArgsForCall()
845 ReplaceLiftedArgNodePlaceholderWithArg(*fbody, original_arg_count, i, in PostprocessLiftedArgsForCall()
850 TF_RETURN_IF_ERROR(GraphToFunctionDef(*fbody->graph, n->type_string(), in PostprocessLiftedArgsForCall()
1097 std::unique_ptr<FunctionBody> fbody; in ExpandHostGraphIntoMainGraph() local
1101 AttrSlice(&attrs), fld, &fbody)); in ExpandHostGraphIntoMainGraph()
1102 Graph* host_graph = fbody->graph; in ExpandHostGraphIntoMainGraph()
[all …]
|
| D | rearrange_function_argument_pass_test.cc | 114 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST() argument
119 *fbody = new_fbody.get(); in TEST()
225 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST() argument
230 *fbody = new_fbody.get(); in TEST()
|
| D | compilability_check_util.cc | 281 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
283 for (const Node* node : fbody->graph->op_nodes()) { in IsCompilableCall()
|
| D | xla_cluster_util.cc | 463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes() local
465 *fbody->graph, lib_runtime, direction, depth + 1, &callee_ref_nodes)); in DoesAnyCalleeHaveRefNodes()
|
| D | encapsulate_subgraphs_pass.cc | 1087 std::unique_ptr<FunctionBody> fbody; in MakePrunedGraphCopyAndInline() local
1089 FunctionDefToBodyHelper(*fdef, node->attrs(), library, &fbody)); in MakePrunedGraphCopyAndInline()
1093 fbody.get(), inline_opts)); in MakePrunedGraphCopyAndInline()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 37 const FunctionBody** fbody) { in GetFunctionBody() argument
43 *fbody = flib_runtime->GetFunctionBody(func_handle); in GetFunctionBody()
71 for (auto fbody : branch_bodies) { in CondConstInputIndices() local
72 TF_RET_CHECK(fbody != nullptr); in CondConstInputIndices()
74 *(fbody->graph), &compile_time_const_arg_indices, in CondConstInputIndices()
97 const FunctionBody* fbody = nullptr; in GetCompileTimeConstInputs() local
99 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "body", &fbody)); in GetCompileTimeConstInputs()
101 TF_RET_CHECK(fbody); in GetCompileTimeConstInputs()
102 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs()
111 *(fbody->graph), &compile_time_const_arg_indices, in GetCompileTimeConstInputs()
[all …]
|
| D | rearrange_function_argument.cc | 310 const FunctionBody* fbody; in MaybeRewriteWhileNode() local
311 TF_RETURN_IF_ERROR(get_function_body_fn(attr_value, &fbody)); in MaybeRewriteWhileNode()
320 for (int i = 0; i < fbody->ret_nodes.size(); i++) { in MaybeRewriteWhileNode()
321 Node* n = fbody->ret_nodes[i]; in MaybeRewriteWhileNode()
350 RearrangeArgNodes(&fbody->arg_nodes, index_mapping); in MaybeRewriteWhileNode()
352 for (int i = 0; i < fbody->ret_nodes.size(); i++) { in MaybeRewriteWhileNode()
353 Node* n = fbody->ret_nodes[i]; in MaybeRewriteWhileNode()
359 fbody->graph->RemoveNode(n); in MaybeRewriteWhileNode()
368 TF_RETURN_IF_ERROR(GraphToFunctionDef(*fbody->graph, new_name, &new_fdef)); in MaybeRewriteWhileNode()
434 const FunctionBody* fbody; in MaybeRewriteIfNode() local
[all …]
|
| D | resource_util.cc | 132 const FunctionBody& fbody, in UpdateResourceUsageFromFunctionBodyAnalysis() argument
143 fbody.graph->BuildNodeNameIndex(); in UpdateResourceUsageFromFunctionBodyAnalysis()
210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp() local
217 fbody->graph, n.type_string(), call_depth + 1, resource_arg_indices, in PropagateThroughCallOp()
221 n, function_name, *fbody, called_function_source_to_path, user_to_source, in PropagateThroughCallOp()
|
| D | xla_compiler.cc | 562 const FunctionBody** fbody) { in GetFunctionBody() argument
567 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
568 TF_RET_CHECK(*fbody); in GetFunctionBody()
573 const FunctionBody** fbody) { in FindFunctionBody() argument
577 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
583 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
592 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph() argument
594 CopyGraph(*fbody->graph, graph.get()); in GetGraph()
690 const FunctionBody* fbody; in CompileFunction() local
691 TF_RETURN_IF_ERROR(FindFunctionBody(fn_name_attrs, &fbody)); in CompileFunction()
[all …]
|
| D | graph_compiler.cc | 247 const FunctionBody* fbody; in CompileFunctionalNode() local
248 TF_RETURN_IF_ERROR(compiler->FindFunctionBody(func, &fbody)); in CompileFunctionalNode()
250 auto graph = compiler->GetGraph(fbody); in CompileFunctionalNode()
|
| D | xla_compiler.h | 449 const FunctionBody** fbody);
453 std::unique_ptr<Graph> GetGraph(const FunctionBody* fbody);
|
| D | tf2xla_util.cc | 187 std::unique_ptr<FunctionBody> fbody; in PropagateConstIntoFuncAttr() local
189 *fdef, AttrSlice(&func_attr.attr()), lookup_fld, &fbody)); in PropagateConstIntoFuncAttr()
192 Graph* func_graph = fbody->graph; in PropagateConstIntoFuncAttr()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 433 std::unique_ptr<FunctionBody>* fbody);
587 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes() local
588 *ret_types = fbody->ret_types; in GetRetTypes()
632 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
633 CHECK_NOTNULL(fbody); in CreateKernel()
640 for (const auto& t : fbody->arg_types) { in CreateKernel()
644 for (const auto& t : fbody->ret_types) { in CreateKernel()
652 &fbody->fdef.signature(), flr, device_->resource_manager(), in CreateKernel()
653 fbody->arg_types, input_memory_types, fbody->ret_types, in CreateKernel()
664 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBody() argument
[all …]
|
| D | inspecting_placer.cc | 130 std::unique_ptr<FunctionBody> fbody; in ComputeIOColocationGroups() local
133 &flib_def_, &fbody)); in ComputeIOColocationGroups()
136 IsolatePlacerInspectionRequiredOps(flib_def_, fbody->graph)); in ComputeIOColocationGroups()
148 fbody->graph, stack_.Push(&node, func.name()), &flib_def_, &device_set_, in ComputeIOColocationGroups()
153 converter.AssignGroups(fbody->arg_nodes, &groups->input_groups); in ComputeIOColocationGroups()
154 converter.AssignGroups(fbody->ret_nodes, &groups->output_groups); in ComputeIOColocationGroups()
|
| D | lower_function_call_op.cc | 93 std::unique_ptr<FunctionBody> fbody; in RewriteFunctionCallNode() local
95 FunctionDefToBodyHelper(*fdef, n->attrs(), &flib_def, &fbody)); in RewriteFunctionCallNode()
98 ValidateInlining(n, fbody.get(), inline_options); in RewriteFunctionCallNode()
101 InlineFunctionBody(flib_def, g, n, fbody.get(), inline_options)); in RewriteFunctionCallNode()
|
| D | function.h | 290 Status ValidateInlining(const Node* node, const FunctionBody* fbody,
302 Node* caller, const FunctionBody* fbody,
389 std::unique_ptr<FunctionBody>* fbody);
398 std::unique_ptr<FunctionBody>* fbody);
|
| D | process_function_library_runtime.cc | 558 std::unique_ptr<FunctionBody> fbody; in GetGraphAndArgRets() local
560 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fdef, attrs, lib_def, &fbody)); in GetGraphAndArgRets()
561 if (!fbody) { in GetGraphAndArgRets()
566 *graph = std::unique_ptr<Graph>(fbody->graph); in GetGraphAndArgRets()
567 arg_nodes->reserve(fbody->arg_nodes.size()); in GetGraphAndArgRets()
568 std::copy(fbody->arg_nodes.begin(), fbody->arg_nodes.end(), in GetGraphAndArgRets()
570 ret_nodes->reserve(fbody->ret_nodes.size()); in GetGraphAndArgRets()
571 std::copy(fbody->ret_nodes.begin(), fbody->ret_nodes.end(), in GetGraphAndArgRets()
573 fbody->graph = nullptr; in GetGraphAndArgRets()
574 ret_node_names->reserve(fbody->ret_nodes.size()); in GetGraphAndArgRets()
[all …]
|
| D | function_test.cc | 305 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface() local
306 FunctionCallFrame frame(fbody->arg_types, fbody->ret_types); in InstantiateAndRunViaCallFrameInterface()
341 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody() local
342 CHECK_NOTNULL(fbody); in GetFuncBody()
344 CopyGraph(*fbody->graph, ret.get()); in GetFuncBody()
357 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody() local
358 CHECK_NOTNULL(fbody); in GetGradBody()
359 std::unique_ptr<FunctionBody> gbody(SymbolicGradient(*fbody)); in GetGradBody()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 856 const FunctionBody& fbody, OutputControlSource output_control_source, in ValidateSideEffectsExecution() argument
863 fbody.graph->nodes(), std::back_inserter(fbody_side_effects), in ValidateSideEffectsExecution()
887 control_sources = {fbody.ret_nodes.begin(), fbody.ret_nodes.end()}; in ValidateSideEffectsExecution()
889 control_sources = {fbody.control_ret_nodes.begin(), in ValidateSideEffectsExecution()
890 fbody.control_ret_nodes.end()}; in ValidateSideEffectsExecution()
907 DFSFrom(*fbody.graph, {side_effect}, /*enter=*/is_control_source, in ValidateSideEffectsExecution()
923 const FunctionBody& fbody) { in ValidateNoDeadOutputs() argument
924 absl::flat_hash_set<const Node*> output_nodes = {fbody.ret_nodes.begin(), in ValidateNoDeadOutputs()
925 fbody.ret_nodes.end()}; in ValidateNoDeadOutputs()
929 for (const Node* n : fbody.graph->nodes()) { in ValidateNoDeadOutputs()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 235 std::unique_ptr<FunctionBody> fbody; in MakeGrapplerFunctionItem() local
237 FunctionDefToBodyHelper(func, func_instantiation_attr, &flib, &fbody)); in MakeGrapplerFunctionItem()
240 fbody->graph->ToGraphDef(&function_body); in MakeGrapplerFunctionItem()
252 const int num_instantiated_inputs = fbody->arg_types.size(); in MakeGrapplerFunctionItem()
253 const int num_instantiated_outputs = fbody->ret_types.size(); in MakeGrapplerFunctionItem()
259 const Node* node = fbody->arg_nodes[in_id]; in MakeGrapplerFunctionItem()
260 const DataType& dtype = fbody->arg_types[in_id]; in MakeGrapplerFunctionItem()
268 const Node* node = fbody->ret_nodes[out_id]; in MakeGrapplerFunctionItem()
269 const DataType& dtype = fbody->ret_types[out_id]; in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 188 StatusOr<mlir::FunctionType> InferLibFunctionType(const FunctionBody& fbody);
194 const FunctionBody& fbody,
1030 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody() argument
1035 arg_nodes->reserve(fbody.arg_nodes.size()); in GetArgsAndRetsFromFunctionBody()
1036 ret_nodes->reserve(fbody.ret_nodes.size()); in GetArgsAndRetsFromFunctionBody()
1037 for (auto arg : fbody.arg_nodes) { in GetArgsAndRetsFromFunctionBody()
1040 for (auto ret : fbody.ret_nodes) { in GetArgsAndRetsFromFunctionBody()
1043 for (const auto& entry : fbody.fdef.resource_arg_unique_id()) { in GetArgsAndRetsFromFunctionBody()
1046 *control_ret_nodes = fbody.control_ret_nodes; in GetArgsAndRetsFromFunctionBody()
1069 std::unique_ptr<FunctionBody> fbody; in ConvertLibFunction() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 314 std::unique_ptr<FunctionBody> fbody; in HashFunction() local
316 &flib_def_, &fbody)); in HashFunction()
317 GraphDef graph_def = fbody->graph->ToGraphDefDebug(); in HashFunction()
323 for (const auto& ret_node : fbody->ret_nodes) { in HashFunction()
332 for (const auto& control_ret_node : fbody->control_ret_nodes) { in HashFunction()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 199 const FunctionBody* fbody; in FunctionDefToGraphDef() local
200 fbody = flib_runtime->GetFunctionBody(handle); in FunctionDefToGraphDef()
201 if (!fbody) { in FunctionDefToGraphDef()
206 CopyGraph(*fbody->graph, graph.get()); in FunctionDefToGraphDef()
|
| /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 …]
|