Home
last modified time | relevance | path

Searched defs:fbody (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc37 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
Dresource_util.cc132 const FunctionBody& fbody, in UpdateResourceUsageFromFunctionBodyAnalysis()
210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp() local
Drearrange_function_argument.cc310 const FunctionBody* fbody; in MaybeRewriteWhileNode() local
436 const FunctionBody* fbody; in MaybeRewriteIfNode() local
525 const FunctionBody* fbody; in RearrangeFunctionArguments() local
Dxla_compiler.cc542 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()
Dgraph_compiler.cc251 const FunctionBody* fbody; in CompileFunctionalNode() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_def_utils.cc33 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBodyHelper()
78 std::unique_ptr<FunctionBody>* fbody) { in FunctionDefToBodyHelper()
Dlower_function_call_op.cc93 std::unique_ptr<FunctionBody> fbody; in RewriteFunctionCallNode() local
Dinline_function_utils.cc285 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
Dfunction.cc593 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
Dinspecting_placer.cc131 std::unique_ptr<FunctionBody> fbody; in ComputeIOColocationGroups() local
/external/tensorflow/tensorflow/compiler/jit/
Dforce_xla_constants_on_host_pass.cc37 const FunctionBody* fbody = nullptr; in Run() local
Dcompilability_check_util.cc361 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()
Drearrange_function_argument_pass_test.cc114 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
225 [&](const NameAttrList &function, const FunctionBody **fbody) { in TEST()
Dxla_kernel_creator.cc95 const FunctionBody* fbody = nullptr; in CreateXlaKernel() local
Dget_compiler_ir.cc59 const FunctionBody* fbody = nullptr; in GetCompilerIr() local
Dextract_outside_compilation_pass.cc834 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
Dxla_cluster_util.cc463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc872 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/
Dmlir.cc117 std::unique_ptr<tensorflow::FunctionBody> fbody; in ImportFunction() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1685 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/
Dfunctions.cc242 std::unique_ptr<FunctionBody> fbody; in MakeGrapplerFunctionItem() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc237 const FunctionBody* fbody; in FunctionDefToGraphDef() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.cc314 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1215 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/
Dhash_utils.cc470 std::unique_ptr<FunctionBody> fbody; in HashFunction() local

12