| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 70 lib_def_ = absl::make_unique<FunctionLibraryDefinition>( in Init()
75 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
82 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::CreateXlaLaunchOpTest
|
| /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::__anon1f3551c50111::Vectorization
372 lib_def_.Find(map_defun_node.attr().at("f").func().name()); in Initialize()
381 return this->lib_def_.LookUpOpDef(op, sig); in Initialize()
385 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(outer_scope, {}, &lib_def_, in Initialize()
393 TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*map_defun_fn, {}, &lib_def_, in Initialize()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 66 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
72 nullptr, env, nullptr, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
78 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
92 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
98 nullptr, env, nullptr, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
104 device_mgr, env, d, graph_def_version, lib_def_, default_thread_pool, in ProcessFunctionLibraryRuntime()
543 options.overlay_lib == nullptr ? lib_def_ : options.overlay_lib; in InstantiateMultiDevice()
838 f->DistributedInit(parent_, function_name, *lib_def_, attrs, options)); in Instantiate()
1008 out_lib_def->reset(new FunctionLibraryDefinition(*lib_def_)); in Clone()
|
| D | function_threadpool_test.cc | 63 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
67 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
194 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anonaa65071b0111::FunctionLibraryRuntimeTest
|
| D | process_function_library_runtime_test.cc | 89 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
93 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
214 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anonbd125a7f0111::ProcessFunctionLibraryRuntimeTest
|
| D | process_function_library_runtime.h | 292 const FunctionLibraryDefinition* lib_def_; variable
|
| D | function_test.cc | 157 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
161 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in Init()
166 fdef_lib_ = lib_def_->ToProto(); in Init()
338 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetFuncBody()
356 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetGradBody()
365 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anona9a56aea0111::FunctionLibraryRuntimeTest
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 154 lib_def_ = in InitFunctionLibraryRuntime()
159 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, lib_def_.get(), in InitFunctionLibraryRuntime()
215 params.flib_def = lib_def_.get(); in CreateSerializationContext()
|
| D | multi_device_iterator_ops.cc | 97 params.function_library = lib_def_; in GetNextFromShard()
116 return lib_def_; in function_library()
362 std::shared_ptr<const FunctionLibraryDefinition> lib_def_ GUARDED_BY(mu_);
|
| D | dataset_test_base.h | 192 std::unique_ptr<FunctionLibraryDefinition> lib_def_; variable
|