| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 38 const FunctionBody** fbody) { in GetFunctionBody()
72 for (auto fbody : branch_bodies) { in CondConstInputIndices() local
97 const FunctionBody* fbody = nullptr; in GetCompileTimeConstInputs() local
152 const FunctionBody* fbody; 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
436 const FunctionBody* fbody; in MaybeRewriteIfNode() local
525 const FunctionBody* fbody; in RearrangeFunctionArguments() local
|
| D | xla_compiler.cc | 547 const FunctionBody** fbody) { in GetFunctionBody()
558 const FunctionBody** fbody, in FindFunctionBody()
584 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph()
745 const FunctionBody* fbody; in CompileFunction() local
1321 [this](const NameAttrList& function, const FunctionBody** fbody) { in CompileGraph()
|
| D | tf2xla_util.cc | 242 std::unique_ptr<FunctionBody> fbody; in PropagateConstIntoFuncAttr() local
325 std::unique_ptr<FunctionBody> fbody; in FindOrInsert() local
419 std::unique_ptr<FunctionBody> fbody; in PropagateConstIntoAndAroundWhileNode() local
|
| D | graph_compiler.cc | 252 const FunctionBody* fbody; in CompileFunctionalNode() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_def_utils.cc | 33 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBodyHelper()
78 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBodyHelper()
|
| D | lower_function_call_op.cc | 83 std::unique_ptr<FunctionBody> fbody; in RewriteFunctionCallNode() local
|
| D | inline_function_utils_test.cc | 80 std::unique_ptr<FunctionBody> fbody; in TEST() local
|
| D | function.cc | 598 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes() local
646 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
680 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBody()
788 std::unique_ptr<FunctionBody> fbody; in Instantiate() local
925 const FunctionBody* fbody; in CreateItem() local
1053 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote() local
1170 const FunctionBody* fbody = GetFunctionBody(handle); in Run() local
1318 const FunctionBody* fbody = GetFunctionBody(handle); in RunSync() local
|
| D | inline_function_utils.cc | 274 Status ValidateNoInline(const FunctionBody* fbody) { in ValidateNoInline()
335 Status ValidateInlining(const Node* node, const FunctionBody* fbody, in ValidateInlining()
479 Node* caller, const FunctionBody* fbody, in InlineFunctionBody()
879 const FunctionBody* fbody = lib->GetFunctionBody(handle); in ExpandInlineFunctions() local
|
| D | inspecting_placer.cc | 131 std::unique_ptr<FunctionBody> fbody; in ComputeIOColocationGroups() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | force_xla_constants_on_host_pass.cc | 37 const FunctionBody* fbody = nullptr; in Run() local
|
| D | compilability_check_util.cc | 344 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
633 const FunctionBody** fbody, in GetBodyAndConstantsAndResources()
667 const tensorflow::FunctionBody* fbody, in GetInputMemoryTypes()
693 const tensorflow::FunctionBody* fbody) { in GetOutputMemoryTypes()
|
| D | xla_kernel_creator.cc | 57 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 | get_compiler_ir.cc | 93 const FunctionBody* fbody = nullptr; in GetCompilerIr() local
|
| D | extract_outside_compilation_pass.cc | 838 std::unique_ptr<FunctionBody> fbody; in PostprocessLiftedArgsForCall() local
1138 std::unique_ptr<FunctionBody> fbody; in ExpandHostGraphIntoMainGraph() local
1216 std::unique_ptr<FunctionBody> fbody; in RewriteShapeInferenceGraph() local
1367 std::unique_ptr<FunctionBody> fbody; in ReplaceKeyPlaceholderWithArgNode() local
1490 std::unique_ptr<FunctionBody> fbody; in AddSendLoopPredToLoopCond() local
2327 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/translate/ |
| D | import_model.cc | 36 mlir::StringRef function_name, const tensorflow::FunctionBody* fbody, in ConvertFunctionToBef()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | function_cache.cc | 71 std::unique_ptr<tensorflow::FunctionBody> fbody; in GetOrAddFunction() local
|
| D | transform_graph_function.cc | 39 std::unique_ptr<FunctionBody>* fbody, in TransformGraphFunction()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 882 const FunctionBody& fbody, OutputControlSource output_control_source, in ValidateSideEffectsExecution()
949 const FunctionBody& fbody) { in ValidateNoDeadOutputs()
1015 std::unique_ptr<FunctionBody>* fbody) { in MakeFunctionBodyForInlining()
1284 std::unique_ptr<FunctionBody> fbody; in InlineFunctionCalls() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1643 std::unique_ptr<FunctionBody> fbody; in InstantiateAssociatedFunction() local
1804 Node* call_node, const FunctionBody& fbody) { in FindArgsToLiftForCallNode()
1839 const FunctionBody& fbody, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
1902 const FunctionBody* fbody) { in RemoveArgsToLiftFromFunctionBody()
2062 const std::vector<DataType>& dtypes, FunctionBody* fbody) { in CleanUpRetvalsForWhileBody()
2242 const FunctionBody* fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgsFromCallNode() local
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.cc | 170 std::unique_ptr<tensorflow::FunctionBody> fbody; in ImportFunction() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 406 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo() local
|