| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 37 const FunctionBody** fbody) { in GetFunctionBody()
71 for (auto fbody : branch_bodies) { in CondConstInputIndices() local
97 const FunctionBody* fbody = nullptr; in GetCompileTimeConstInputs() local
|
| D | resource_util.cc | 132 const FunctionBody& fbody, in UpdateResourceUsageFromFunctionBodyAnalysis()
210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp() local
|
| D | rearrange_function_argument.cc | 310 const FunctionBody* fbody; in MaybeRewriteWhileNode() local
434 const FunctionBody* fbody; in MaybeRewriteIfNode() local
518 const FunctionBody* fbody; in RearrangeFunctionArguments() local
|
| D | xla_compiler.cc | 562 const FunctionBody** fbody) { in GetFunctionBody()
573 const FunctionBody** fbody) { in FindFunctionBody()
592 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph()
690 const FunctionBody* fbody; in CompileFunction() local
1301 [this](const NameAttrList& function, const FunctionBody** fbody) { in CompileGraph()
|
| D | graph_compiler.cc | 247 const FunctionBody* fbody; in CompileFunctionalNode() local
|
| D | tf2xla_util.cc | 187 std::unique_ptr<FunctionBody> fbody; in PropagateConstIntoFuncAttr() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_util.cc | 87 const FunctionBody** fbody, in GetBodyAndConstantsAndResources()
164 const FunctionBody* fbody = nullptr; in CreateXlaKernel() local
|
| D | rearrange_function_argument_pass_test.cc | 114 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
225 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
|
| D | extract_outside_compilation_pass.cc | 807 std::unique_ptr<FunctionBody> fbody; in PostprocessLiftedArgsForCall() local
1097 std::unique_ptr<FunctionBody> fbody; in ExpandHostGraphIntoMainGraph() local
1175 std::unique_ptr<FunctionBody> fbody; in RewriteShapeInferenceGraph() local
1318 std::unique_ptr<FunctionBody> fbody; in ReplaceKeyPlaceholderWithArgNode() local
1443 std::unique_ptr<FunctionBody> fbody; in AddSendLoopPredToLoopCond() local
2273 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction() local
|
| D | compilability_check_util.cc | 281 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
|
| D | xla_cluster_util.cc | 463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes() local
|
| D | encapsulate_subgraphs_pass.cc | 1087 std::unique_ptr<FunctionBody> fbody; in MakePrunedGraphCopyAndInline() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_function_call_op.cc | 93 std::unique_ptr<FunctionBody> fbody; in RewriteFunctionCallNode() local
|
| D | function.cc | 587 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes() local
632 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
664 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBody()
765 std::unique_ptr<FunctionBody> fbody; in Instantiate() local
927 const FunctionBody* fbody; in CreateItem() local
1036 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote() local
1153 const FunctionBody* fbody = GetFunctionBody(handle); in Run() local
1670 Status ValidateNoInline(const FunctionBody* fbody) { in ValidateNoInline()
1731 Status ValidateInlining(const Node* node, const FunctionBody* fbody, in ValidateInlining()
1875 Node* caller, const FunctionBody* fbody, in InlineFunctionBody()
[all …]
|
| D | inspecting_placer.cc | 130 std::unique_ptr<FunctionBody> fbody; in ComputeIOColocationGroups() local
|
| D | process_function_library_runtime.cc | 558 std::unique_ptr<FunctionBody> fbody; in GetGraphAndArgRets() local
923 const FunctionBody* fbody = target_flr->GetFunctionBody(comp_data.handle_); in GetOutputDevices() local
|
| D | function_test.cc | 305 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface() local
341 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody() local
357 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 856 const FunctionBody& fbody, OutputControlSource output_control_source, in ValidateSideEffectsExecution()
923 const FunctionBody& fbody) { in ValidateNoDeadOutputs()
989 std::unique_ptr<FunctionBody>* fbody) { in MakeFunctionBodyForInlining()
1240 std::unique_ptr<FunctionBody> fbody; in InlineFunctionCalls() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 235 std::unique_ptr<FunctionBody> fbody; in MakeGrapplerFunctionItem() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 199 const FunctionBody* fbody; in FunctionDefToGraphDef() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1030 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1069 std::unique_ptr<FunctionBody> fbody; in ConvertLibFunction() local
1716 const FunctionBody& fbody) { in InferLibFunctionType()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 314 std::unique_ptr<FunctionBody> fbody; in HashFunction() local
|