| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 36 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() function
44 *fbody = flib_runtime->GetFunctionBody(func_handle); in GetFunctionBody()
58 fbodies->push_back(flib_runtime->GetFunctionBody(func_handle)); in GetFunctionBodies()
98 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "cond", &fcond)); in GetCompileTimeConstInputs()
99 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "body", &fbody)); in GetCompileTimeConstInputs()
140 GetFunctionBody(flib_runtime, node, "then_branch", &fthen)); in GetCompileTimeConstInputs()
142 GetFunctionBody(flib_runtime, node, "else_branch", &felse)); in GetCompileTimeConstInputs()
153 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "f", &fbody)); in GetCompileTimeConstInputs()
|
| D | functionalize_control_flow.cc | 220 const FunctionBody* body = flr->GetFunctionBody(handle); in FunctionalizeControlFlowForFunction()
|
| D | resource_util.cc | 210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp()
|
| D | xla_compiler.cc | 545 static Status GetFunctionBody(const NameAttrList& function, in GetFunctionBody() function
552 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
563 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
569 GetFunctionBody(function, flib_runtime_, fbody), in FindFunctionBody()
|
| D | tf2xla_util_test.cc | 316 const FunctionBody* body = flr->GetFunctionBody(first_handle); in TEST()
|
| D | functionalize_control_flow_test.cc | 296 const FunctionBody* body = flr->GetFunctionBody(handle); in RunTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 163 const FunctionBody* GetFunctionBody(Handle h) override;
227 const FunctionBody* FunctionLibraryRuntimeOverlay::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeOverlay
228 return base_flr_->GetFunctionBody(h); in GetFunctionBody()
344 const FunctionBody* GetFunctionBody(Handle handle) override;
575 const FunctionBody* FunctionLibraryRuntimeImpl::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeImpl
598 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes()
646 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel()
718 const FunctionBody* f_body = GetFunctionBody(f_handle); in InstantiateSymbolicGradient()
1053 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote()
1170 const FunctionBody* fbody = GetFunctionBody(handle); in Run()
[all …]
|
| D | function_test.cc | 290 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface()
326 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody()
342 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody()
401 const FunctionBody* func_body = flr0_->GetFunctionBody(handle); in TEST_F()
|
| D | inline_function_utils.cc | 879 const FunctionBody* fbody = lib->GetFunctionBody(handle); in ExpandInlineFunctions()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.cc | 344 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall()
639 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
|
| D | xla_cluster_util.cc | 466 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes()
|
| D | extract_outside_compilation_pass.cc | 2327 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 431 const FunctionBody* body = flr->GetFunctionBody(handle); in DetectHostTrainingLoop()
|
| D | encapsulate_tpu_computations_pass.cc | 2242 const FunctionBody* fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgsFromCallNode()
2409 const FunctionBody* function_fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgs()
|
| D | distributed_tpu_rewrite_pass.cc | 1883 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetComputationForTPUReplicateOp()
2068 const FunctionBody* body = flr->GetFunctionBody(handle); in InferXlaShardingFromNeighbors()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 406 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 814 virtual const FunctionBody* GetFunctionBody(Handle h) = 0;
|
| /external/swiftshader/third_party/SPIRV-Tools/source/diff/ |
| D | diff.cpp | 454 InstructionList GetFunctionBody(opt::IRContext* context,
1553 InstructionList Differ::GetFunctionBody(opt::IRContext* context, in GetFunctionBody() function in spvtools::diff::__anon5a2be8a60111::Differ
1599 InstructionList body = GetFunctionBody(context, function); in GetFunctionBodies()
|
| /external/angle/third_party/spirv-tools/src/source/diff/ |
| D | diff.cpp | 463 InstructionList GetFunctionBody(opt::IRContext* context,
1595 InstructionList Differ::GetFunctionBody(opt::IRContext* context, in GetFunctionBody() function in spvtools::diff::__anonf81e29200111::Differ
1641 InstructionList body = GetFunctionBody(context, function); in GetFunctionBodies()
|
| /external/deqp-deps/SPIRV-Tools/source/diff/ |
| D | diff.cpp | 454 InstructionList GetFunctionBody(opt::IRContext* context,
1553 InstructionList Differ::GetFunctionBody(opt::IRContext* context, in GetFunctionBody() function in spvtools::diff::__anonf938b5ac0111::Differ
1599 InstructionList body = GetFunctionBody(context, function); in GetFunctionBodies()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 191 ctx->function_library()->GetFunctionBody(fn_handle); in CreateShortCircuitInfo()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 303 fbody = flib_runtime->GetFunctionBody(handle); in FunctionDefToGraphDef()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_functional_ops.cc | 2298 const FunctionBody* fbody = library_runtime_->GetFunctionBody(handle); in GetGraphFromFunction()
|