| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 55 lib_def_(lib_def), in RemoteExecuteNode()
94 return RunShapeInference(ndef_, *lib_def_, inputs_, retvals_); in Prepare()
133 const FunctionLibraryDefinition* lib_def_; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 73 lib_def_ = absl::make_unique<FunctionLibraryDefinition>( in Init()
79 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, in Init()
86 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::XlaKernelCreatorTest
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 217 return lib_def_; in GetFunctionLibraryDefinition()
271 lib_def_(std::move(lib_def)), in MultiDeviceFunctionData()
282 FunctionLibraryDefinition lib_def_; member
450 const FunctionLibraryDefinition* lib_def_; variable
|
| D | function_threadpool_test.cc | 62 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
67 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, default_thread_pool, in Init()
198 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anonbf8091390111::FunctionLibraryRuntimeTest
|
| D | process_function_library_runtime.cc | 97 lib_def_(lib_def), in ProcessFunctionLibraryRuntime()
109 graph_def_version, lib_def_, default_thread_pool, optimizer_options, in ProcessFunctionLibraryRuntime()
227 graph_def_version_, lib_def_, default_thread_pool_, in InitializeDeviceAndFlr()
480 !(dtype == DT_RESOURCE && IsFunctionCall(*lib_def_, *src_node)); in PinArgsAndRets()
701 options.lib_def == nullptr ? lib_def_ : options.lib_def; in InstantiateMultiDevice()
769 *dev_set, options.config_proto, &graph, &data->lib_def_, in InstantiateMultiDevice()
792 optimization_options.flib_def = &data->lib_def_; in InstantiateMultiDevice()
824 &data->lib_def_, *dev_set, cpu_device, &graph); in InstantiateMultiDevice()
911 FunctionLibraryDefinition* data_lib_def = &data->lib_def_; in InstantiateMultiDevice()
1269 options.lib_def == nullptr ? *lib_def_ : *options.lib_def, attrs, options, in InstantiateRemote()
[all …]
|
| D | function.cc | 150 : base_flr_(base_flr), lib_def_(lib_def) {} in FunctionLibraryRuntimeOverlay()
200 const FunctionLibraryDefinition* lib_def_; // not owned member in tensorflow::FunctionLibraryRuntimeOverlay
210 if (!options.lib_def && lib_def_) { in Instantiate()
212 options_copy.lib_def = lib_def_; in Instantiate()
273 const Status s = lib_def_->LookUpOpDef(function_name, &op_def); in IsStateful()
300 return lib_def_ ? lib_def_ : base_flr_->GetFunctionLibraryDefinition(); in GetFunctionLibraryDefinition()
|
| D | function_test.cc | 159 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
164 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, /*thread_pool=*/nullptr, in Init()
174 fdef_lib_ = lib_def_->ToProto(); in Init()
345 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetFuncBody()
363 std::unique_ptr<Graph> ret(new Graph(lib_def_.get())); in GetGradBody()
372 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anon196ffbc80111::FunctionLibraryRuntimeTest
413 TF_CHECK_OK(lib_def_->AddFunctionDef(func, stack_traces)); in TEST_F()
|
| D | process_function_library_runtime_test.cc | 136 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto)); in Init()
141 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, in Init()
326 std::unique_ptr<FunctionLibraryDefinition> lib_def_; member in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest
|
| /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::__anoncb11dea30111::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 | 95 const FunctionLibraryDefinition* lib_def() const { return lib_def_.get(); } in lib_def()
120 std::unique_ptr<FunctionLibraryDefinition> lib_def_ = nullptr; variable
|
| D | dataset_test_base.cc | 398 lib_def_ = in InitFunctionLibraryRuntime()
404 TF_GRAPH_DEF_VERSION, lib_def_.get(), opts, thread_pool_.get(), in InitFunctionLibraryRuntime()
|
| D | dataset_test_base.h | 767 std::unique_ptr<FunctionLibraryDefinition> lib_def_; variable
|
| D | captured_function.cc | 506 (*out_metadata)->func_.name(), &(*out_metadata)->lib_def_)); in Create()
|