| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 35 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() function
43 *fbody = flib_runtime->GetFunctionBody(func_handle); in GetFunctionBody()
57 fbodies->push_back(flib_runtime->GetFunctionBody(func_handle)); in GetFunctionBodies()
97 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "cond", &fcond)); in GetCompileTimeConstInputs()
98 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "body", &fbody)); in GetCompileTimeConstInputs()
138 GetFunctionBody(flib_runtime, node, "then_branch", &fthen)); in GetCompileTimeConstInputs()
140 GetFunctionBody(flib_runtime, node, "else_branch", &felse)); in GetCompileTimeConstInputs()
151 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "f", &fbody)); in GetCompileTimeConstInputs()
|
| D | resource_util.cc | 210 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in PropagateThroughCallOp()
|
| D | functionalize_control_flow.cc | 221 const FunctionBody* body = flr->GetFunctionBody(handle); in FunctionalizeControlFlowForFunction()
|
| D | xla_compiler.cc | 540 static Status GetFunctionBody(const NameAttrList& function, in GetFunctionBody() function
547 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
558 auto status = GetFunctionBody(function, local_flib_runtime_, fbody); in FindFunctionBody()
564 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 | 277 const FunctionBody* body = flr->GetFunctionBody(handle); in RunTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 159 const FunctionBody* GetFunctionBody(Handle h) override;
223 const FunctionBody* FunctionLibraryRuntimeOverlay::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeOverlay
224 return base_flr_->GetFunctionBody(h); in GetFunctionBody()
340 const FunctionBody* GetFunctionBody(Handle handle) override;
570 const FunctionBody* FunctionLibraryRuntimeImpl::GetFunctionBody(Handle h) { in GetFunctionBody() function in tensorflow::FunctionLibraryRuntimeImpl
593 const FunctionBody* fbody = GetFunctionBody(h); in GetRetTypes()
641 const FunctionBody* fbody = GetFunctionBody(handle); in CreateKernel()
713 const FunctionBody* f_body = GetFunctionBody(f_handle); in InstantiateSymbolicGradient()
1018 const FunctionBody* fbody = GetFunctionBody(handle); in RunRemote()
1135 const FunctionBody* fbody = GetFunctionBody(handle); in Run()
[all …]
|
| D | function_test.cc | 307 const FunctionBody* fbody = flr->GetFunctionBody(handle); in InstantiateAndRunViaCallFrameInterface()
343 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetFuncBody()
359 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetGradBody()
418 const FunctionBody* func_body = flr0_->GetFunctionBody(handle); in TEST_F()
|
| D | inline_function_utils.cc | 870 const FunctionBody* fbody = lib->GetFunctionBody(handle); in ExpandInlineFunctions()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util.cc | 361 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in IsCompilableCall()
629 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
|
| D | xla_cluster_util.cc | 463 const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle); in DoesAnyCalleeHaveRefNodes()
|
| D | extract_outside_compilation_pass.cc | 2326 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 418 const FunctionBody* body = flr->GetFunctionBody(handle); in DetectHostTrainingLoop()
|
| D | encapsulate_tpu_computations_pass.cc | 2288 const FunctionBody* fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgsFromCallNode()
2455 const FunctionBody* function_fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgs()
|
| D | distributed_tpu_rewrite_pass.cc | 1670 const FunctionBody* fbody = flr->GetFunctionBody(handle); in GetComputationForTPUReplicateOp()
1855 const FunctionBody* body = flr->GetFunctionBody(handle); in InferXlaShardingFromNeighbors()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 758 virtual const FunctionBody* GetFunctionBody(Handle h) = 0;
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 314 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 238 fbody = flib_runtime->GetFunctionBody(handle); in FunctionDefToGraphDef()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 190 ctx->function_library()->GetFunctionBody(fn_handle); in CreateShortCircuitInfo()
|