| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 37 const FunctionBody** fbody) { in GetFunctionBody()
71 for (auto fbody : branch_bodies) { in CondConstInputIndices() local
96 const FunctionBody* fbody = nullptr; in GetCompileTimeConstInputs() local
150 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 | 542 const FunctionBody** fbody) { in GetFunctionBody()
553 const FunctionBody** fbody, in FindFunctionBody()
579 std::unique_ptr<Graph> XlaCompiler::GetGraph(const FunctionBody* fbody) { in GetGraph()
736 const FunctionBody* fbody; in CompileFunction() local
1291 [this](const NameAttrList& function, const FunctionBody** fbody) { in CompileGraph()
|
| D | graph_compiler.cc | 251 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 | 93 std::unique_ptr<FunctionBody> fbody; in RewriteFunctionCallNode() local
|
| D | inline_function_utils.cc | 285 Status ValidateNoInline(const FunctionBody* fbody) { in ValidateNoInline()
346 Status ValidateInlining(const Node* node, const FunctionBody* fbody, in ValidateInlining()
490 Node* caller, const FunctionBody* fbody, in InlineFunctionBody()
870 const FunctionBody* fbody = lib->GetFunctionBody(handle); in ExpandInlineFunctions() local
|
| D | function.cc | 593 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes() local
641 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel() local
675 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBody()
776 std::unique_ptr<FunctionBody> fbody; in Instantiate() local
907 const FunctionBody* fbody; in CreateItem() local
1018 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote() local
1135 const FunctionBody* fbody = GetFunctionBody(handle); in Run() local
1281 const FunctionBody* fbody = GetFunctionBody(handle); in RunSync() 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 | 361 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall() local
623 const FunctionBody** fbody, in GetBodyAndConstantsAndResources()
657 const tensorflow::FunctionBody* fbody, in GetInputMemoryTypes()
683 const tensorflow::FunctionBody* fbody) { in GetOutputMemoryTypes()
|
| 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 | xla_kernel_creator.cc | 95 const FunctionBody* fbody = nullptr; in CreateXlaKernel() local
|
| D | get_compiler_ir.cc | 59 const FunctionBody* fbody = nullptr; in GetCompilerIr() local
|
| D | extract_outside_compilation_pass.cc | 834 std::unique_ptr<FunctionBody> fbody; in PostprocessLiftedArgsForCall() local
1132 std::unique_ptr<FunctionBody> fbody; in ExpandHostGraphIntoMainGraph() local
1210 std::unique_ptr<FunctionBody> fbody; in RewriteShapeInferenceGraph() local
1353 std::unique_ptr<FunctionBody> fbody; in ReplaceKeyPlaceholderWithArgNode() local
1479 std::unique_ptr<FunctionBody> fbody; in AddSendLoopPredToLoopCond() local
2326 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction() local
|
| D | xla_cluster_util.cc | 463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 872 const FunctionBody& fbody, OutputControlSource output_control_source, in ValidateSideEffectsExecution()
939 const FunctionBody& fbody) { in ValidateNoDeadOutputs()
1005 std::unique_ptr<FunctionBody>* fbody) { in MakeFunctionBodyForInlining()
1274 std::unique_ptr<FunctionBody> fbody; in InlineFunctionCalls() local
|
| /external/tensorflow/tensorflow/compiler/mlir/python/ |
| D | mlir.cc | 117 std::unique_ptr<tensorflow::FunctionBody> fbody; in ImportFunction() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1685 std::unique_ptr<FunctionBody> fbody; in InstantiateAssociatedFunction() local
1846 Node* call_node, const FunctionBody& fbody) { in FindArgsToLiftForCallNode()
1881 const FunctionBody& fbody, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
1946 const FunctionBody* fbody) { in RemoveArgsToLiftFromFunctionBody()
2108 const std::vector<DataType>& dtypes, FunctionBody* fbody) { in CleanUpRetvalsForWhileBody()
2288 const FunctionBody* fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgsFromCallNode() local
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions.cc | 242 std::unique_ptr<FunctionBody> fbody; in MakeGrapplerFunctionItem() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 237 const FunctionBody* fbody; in FunctionDefToGraphDef() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 314 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1215 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1249 std::unique_ptr<FunctionBody> fbody; in ConvertLibFunction() local
2091 const FunctionBody& fbody) { in InferLibFunctionType()
3903 const FunctionBody* fbody, const FunctionLibraryDefinition& flib_def, in ConvertFunctionToMlir()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | hash_utils.cc | 470 std::unique_ptr<FunctionBody> fbody; in HashFunction() local
|