| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 49 lib_def_(lib_def), in RemoteExecuteNode()
78 return RunShapeInference(ndef_, *lib_def_, inputs_, retvals_); in Prepare()
101 const FunctionLibraryDefinition* lib_def_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 71 lib_def_ = absl::make_unique<FunctionLibraryDefinition>( in Init()
77 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, in Init()
84 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::XlaKernelCreatorTest
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 197 return lib_def_; in GetFunctionLibraryDefinition()
261 lib_def_(std::move(lib_def)), in MultiDeviceFunctionData()
271 FunctionLibraryDefinition lib_def_; member
440 const FunctionLibraryDefinition* lib_def_; variable
|
| D | function_threadpool_test.cc | 63 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
68 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, default_thread_pool)); in Init()
193 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anon6d08b3dd0111::FunctionLibraryRuntimeTest
|
| D | process_function_library_runtime.cc | 90 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
99 graph_def_version, lib_def_, default_thread_pool, optimizer_options, in ProcessFunctionLibraryRuntime()
106 lib_def_, default_thread_pool, optimizer_options, custom_kernel_creator, in ProcessFunctionLibraryRuntime()
433 !(dtype == DT_RESOURCE && IsFunctionCall(*lib_def_, *src_node)); in PinArgsAndRets()
623 options.lib_def == nullptr ? lib_def_ : options.lib_def; in InstantiateMultiDevice()
682 optimization_options.flib_def = &data->lib_def_; in InstantiateMultiDevice()
709 &data->lib_def_, device_set_, cpu_device, &graph); in InstantiateMultiDevice()
790 FunctionLibraryDefinition* data_lib_def = &data->lib_def_; in InstantiateMultiDevice()
1146 options.lib_def == nullptr ? *lib_def_ : *options.lib_def, attrs, options, in InstantiateRemote()
1452 lib_def_->default_registry(), FunctionDefLibrary{}); in Clone()
[all …]
|
| D | function.cc | 171 : base_flr_(base_flr), lib_def_(lib_def) {} in FunctionLibraryRuntimeOverlay()
214 const FunctionLibraryDefinition* lib_def_; // not owned member in tensorflow::FunctionLibraryRuntimeOverlay
224 if (!options.lib_def && lib_def_) { in Instantiate()
226 options_copy.lib_def = lib_def_; in Instantiate()
275 const Status s = lib_def_->LookUpOpDef(function_name, &op_def); in IsStateful()
302 return lib_def_ ? lib_def_ : base_flr_->GetFunctionLibraryDefinition(); in GetFunctionLibraryDefinition()
|
| D | process_function_library_runtime_test.cc | 117 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
122 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, nullptr, cluster_flr_.get(), in Init()
287 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anon7df932810111::ProcessFunctionLibraryRuntimeTest
|
| D | function_test.cc | 163 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
168 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts)); in Init()
172 fdef_lib_ = lib_def_->ToProto(); in Init()
343 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetFuncBody()
361 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetGradBody()
370 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anond2259cec0111::FunctionLibraryRuntimeTest
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 137 : lib_(lib), lib_def_(OpRegistry::Global(), *lib) {} in Vectorization()
214 FunctionLibraryDefinition lib_def_; member in tensorflow::grappler::vectorization_utils::__anonfb8d47c70111::Vectorization
372 lib_def_.Find(map_defun_node.attr().at("f").func().name()); in Initialize()
382 FunctionDefToBodyHelper(outer_scope, {}, &lib_def_, &outer_fn)); in Initialize()
388 FunctionDefToBodyHelper(*map_defun_fn, {}, &lib_def_, &map_defun_fn_)); in Initialize()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.h | 154 const FunctionLibraryDefinition* lib_def() const { return lib_def_.get(); } in lib_def()
179 std::unique_ptr<FunctionLibraryDefinition> lib_def_ = nullptr; variable
|
| D | captured_function.cc | 448 (*out_metadata)->func_.name(), &(*out_metadata)->lib_def_)); in Create()
456 const FunctionDef* fdef = lib_def_->Find(func_.name()); in ValidateMultiDevice()
|
| D | dataset_test_base.cc | 391 lib_def_ = in InitFunctionLibraryRuntime()
397 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, thread_pool_.get(), in InitFunctionLibraryRuntime()
|
| D | dataset_test_base.h | 720 std::unique_ptr<FunctionLibraryDefinition> lib_def_; variable
|