Home
last modified time | relevance | path

Searched refs:GetFunctionBody (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc148 const FunctionBody* fcond = flib_runtime->GetFunctionBody(cond_handle); in GetCompileTimeConstInputs()
149 const FunctionBody* fbody = flib_runtime->GetFunctionBody(body_handle); in GetCompileTimeConstInputs()
Dfunctionalize_control_flow.cc113 const FunctionBody* body = flr->GetFunctionBody(handle); in FunctionalizeControlFlowForFunction()
Dxla_compiler.cc469 static Status GetFunctionBody(const NameAttrList& function, in GetFunctionBody() function
476 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
486 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
492 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
Dtf2xla_util_test.cc311 const FunctionBody* body = flr->GetFunctionBody(first_handle); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc180 const FunctionBody* GetFunctionBody(Handle h) override;
232 const FunctionBody* FunctionLibraryRuntimeOverlay::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeOverlay
233 return base_flr_->GetFunctionBody(h); in GetFunctionBody()
323 const FunctionBody* GetFunctionBody(Handle handle) override;
518 const FunctionBody* FunctionLibraryRuntimeImpl::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeImpl
571 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel()
639 const FunctionBody* f_body = GetFunctionBody(f_handle); in InstantiateSymbolicGradient()
970 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote()
1087 const FunctionBody* fbody = GetFunctionBody(handle); in Run()
1814 const FunctionBody* fbody = lib->GetFunctionBody(handle); in ExpandInlineFunctions()
Dfunction_test.cc300 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface()
336 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody()
352 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody()
Dprocess_function_library_runtime.cc717 const FunctionBody* fbody = target_flr->GetFunctionBody(comp_data.handle_); in GetOutputDevices()
/external/tensorflow/tensorflow/compiler/jit/
Dcreate_xla_launch_op.cc124 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
Dmark_for_compilation_pass.cc240 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall()
Dextract_outside_compilation_pass.cc1604 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc47 ctx->function_library()->GetFunctionBody(fn_handle); in ComputeShortCircuitIndices()
Dcaptured_function.cc146 const FunctionBody* fbody = lib->GetFunctionBody(f_handle); in Instantiate()
/external/tensorflow/tensorflow/core/framework/
Dfunction.h602 virtual const FunctionBody* GetFunctionBody(Handle h) = 0;