| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 111 ndef, flr_->GetFunctionLibraryDefinition(), &props)); in Init()
143 ->GetFunctionLibraryDefinition() in InstantiateFunc()
146 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in InstantiateFunc()
167 pflr_->GetFunctionLibraryDefinition(); in InstantiateFunc()
|
| D | execute.cc | 334 ctx.pflr()->GetFunctionLibraryDefinition()->Find(op->Name()); in GetFuncAttr()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.cc | 173 if (IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)) { in Compile()
215 if (flib.GetFunctionLibraryDefinition()->Find(node.def().op())) { in GetFunctionNameAndAttr()
228 TF_RET_CHECK(IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)); in CompileFunctionalNode()
|
| D | resource_util.cc | 300 if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), *n)) { in AnalyzeResourceUsage()
|
| D | xla_compiler.cc | 643 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
666 flib_runtime_->GetFunctionLibraryDefinition(), &shape_info) in GetGraph()
1295 pflr_->GetFunctionLibraryDefinition())); in CompileGraph()
1300 flib_runtime_->GetFunctionLibraryDefinition()); in CompileGraph()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_segment.cc | 107 lib->GetFunctionLibraryDefinition()->Find(node_op) == nullptr && in ShouldOwnKernel()
|
| D | function.h | 867 virtual const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 141 options.flib_def = function_library.GetFunctionLibraryDefinition(); in GenerateCompilerOptions()
|
| D | xla_cluster_util.cc | 396 *lib_runtime->GetFunctionLibraryDefinition(); in GetCallTargetListFromNode()
|
| D | compilability_check_util.cc | 436 if (IsFunctionCall(*lib_runtime->GetFunctionLibraryDefinition(), node)) { in IsCompilableNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 180 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
299 FunctionLibraryRuntimeOverlay::GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeOverlay
300 return lib_def_ ? lib_def_ : base_flr_->GetFunctionLibraryDefinition(); in GetFunctionLibraryDefinition()
358 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() in GetFunctionLibraryDefinition() function in tensorflow::FunctionLibraryRuntimeImpl
623 flr->GetFunctionLibraryDefinition(); in CreateKernel()
919 flr->GetFunctionLibraryDefinition(); in CreateItem()
|
| D | inline_function_utils.cc | 851 const FunctionLibraryDefinition* fld = lib->GetFunctionLibraryDefinition(); in ExpandInlineFunctions()
855 if (!IsFunctionCall(*lib->GetFunctionLibraryDefinition(), *node)) { in ExpandInlineFunctions()
|
| D | process_function_library_runtime.h | 216 const FunctionLibraryDefinition* GetFunctionLibraryDefinition() const { in GetFunctionLibraryDefinition() function
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 107 flib_runtime->GetFunctionLibraryDefinition(); in HostComputeOp()
187 ctx, ctx->function_library()->GetFunctionLibraryDefinition(), in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 142 node_def_, flr->GetFunctionLibraryDefinition(), &props); in InitOpWithFunctionLibrary()
|
| D | trt_engine_op.cc | 236 flib_runtime->GetFunctionLibraryDefinition(); in FunctionDefToGraphDef()
288 lib->GetFunctionLibraryDefinition()->Find(func_.name()); in ConstructFunctionHandle()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | partitioned_function_ops.cc | 137 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices()
|
| D | batch_kernels.cc | 339 flib_->GetFunctionLibraryDefinition()->Find(func_.name()); in InstantiateFunction()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 483 flr != nullptr ? flr->GetFunctionLibraryDefinition() : nullptr, in RunShapeInferenceOnComputation()
608 flib_runtime->GetFunctionLibraryDefinition(), in CompileLocallyAndFillHostCache()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 432 params_.function_library->GetFunctionLibraryDefinition(), &props); in LegalizeOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 505 ctx->function_library()->GetFunctionLibraryDefinition(), in Create()
|
| D | iterator_ops.cc | 543 *ctx->function_library()->GetFunctionLibraryDefinition()); in CreatePrivateFLR()
|
| D | dataset_test_base.cc | 310 node_def, flr_->GetFunctionLibraryDefinition(), &props)); in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1849 *call_edge->dst(), flr->GetFunctionLibraryDefinition()); in InferXlaShardingFromNeighbors()
4472 flr->GetFunctionLibraryDefinition(), in Run()
|