| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | kernel_and_device.cc | 112 ndef, flr_->GetFunctionLibraryDefinition(), &props)); in Init()
149 ->GetFunctionLibraryDefinition() in InstantiateFunc()
152 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in InstantiateFunc()
173 pflr_->GetFunctionLibraryDefinition(); in InstantiateFunc()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | graph_compiler.cc | 174 if (IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)) { in Compile()
216 if (flib.GetFunctionLibraryDefinition()->Find(node.def().op())) { in GetFunctionNameAndAttr()
229 TF_RET_CHECK(IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)); in CompileFunctionalNode()
|
| H A D | mlir_xla_op_kernel.cc | 108 *ctx->function_library()->GetFunctionLibraryDefinition(), debug_info, in ConstructXlaOp()
|
| H A D | xla_compiler.cc | 648 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
671 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
1325 pflr_->GetFunctionLibraryDefinition())); in CompileGraph()
1330 flib_runtime_->GetFunctionLibraryDefinition()); in CompileGraph()
|
| H A D | const_analysis.cc | 119 flib_runtime->GetFunctionLibraryDefinition())); in GetCompileTimeConstInputs()
|
| H A D | resource_util.cc | 300 if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), *n)) { in AnalyzeResourceUsage()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function.cc | 184 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
303 FunctionLibraryRuntimeOverlay::GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeOverlay
304 return lib_def_ ? lib_def_ : base_flr_->GetFunctionLibraryDefinition(); in GetFunctionLibraryDefinition()
362 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeImpl
628 flr->GetFunctionLibraryDefinition(); in CreateKernel()
937 flr->GetFunctionLibraryDefinition(); in CreateItem()
|
| H A D | inline_function_utils.cc | 860 const FunctionLibraryDefinition* fld = lib->GetFunctionLibraryDefinition(); in ExpandInlineFunctions()
864 if (!IsFunctionCall(*lib->GetFunctionLibraryDefinition(), *node)) { in ExpandInlineFunctions()
|
| H A D | process_function_library_runtime.h | 224 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_segment.cc | 107 lib->GetFunctionLibraryDefinition()->Find(node_op) == nullptr && in ShouldOwnKernel()
|
| H A D | function.h | 929 virtual const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | host_compute_ops.cc | 119 flib_runtime->GetFunctionLibraryDefinition(); in HostComputeOp()
201 ctx, ctx->function_library()->GetFunctionLibraryDefinition(), in Compile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_platform_info.cc | 178 options.flib_def = function_library.GetFunctionLibraryDefinition(); in GenerateCompilerOptions()
|
| H A D | xla_cluster_util.cc | 399 *lib_runtime->GetFunctionLibraryDefinition(); in GetCallTargetListFromNode()
|
| H A D | compilability_check_util.cc | 421 if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), node)) { in IsCompilableNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | op_kernel_runner.cc | 72 ndef, flr->GetFunctionLibraryDefinition(), &props)); in CreateOpKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | op_kernel_runner.cc | 74 ndef, flr->GetFunctionLibraryDefinition(), &props)); in CreateOpKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| H A D | rpc_ops.cc | 657 instantiate_opts.lib_def = lib->GetFunctionLibraryDefinition(); in Compute()
662 lib->GetFunctionLibraryDefinition()->Find(func_.name()); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | partitioned_function_ops.cc | 138 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op_test.cc | 181 node_def_, flr->GetFunctionLibraryDefinition(), &props); in InitOpWithFunctionLibrary()
|
| H A D | trt_engine_op.cc | 301 flib_runtime->GetFunctionLibraryDefinition(); in FunctionDefToGraphDef()
353 lib->GetFunctionLibraryDefinition()->Find(func_.name()); in ConstructFunctionHandle()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_compile_op_common.cc | 200 flib_runtime->GetFunctionLibraryDefinition(), in CompileLocallyAndFillHostCacheInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_compile.cc | 366 flr != nullptr ? flr->GetFunctionLibraryDefinition() : nullptr, in RunShapeInferenceOnComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | while_op.cc | 124 ctx->compiler()->flib_runtime()->GetFunctionLibraryDefinition(); in GetLoopInvariants()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels.cc | 336 cc_ctx->function_library()->GetFunctionLibraryDefinition()->ToProto(); in TF_GetSerializedFunctionDefLibrary()
|