Searched refs:flr_ (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op_test.cc | 77 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
80 FunctionLibraryRuntime* flr_; member in tensorflow::CreateXlaLaunchOpTest
100 flr_, ToNodeDef(R"pb( in TEST_F()
123 Status status = CreateXlaLaunchOp(flr_, ToNodeDef(R"proto( in TEST_F()
137 Status status = CreateXlaLaunchOp(flr_, ToNodeDef(R"proto( in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 54 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
55 CHECK(flr_ != nullptr);
58 FunctionLibraryRuntime* function_library_runtime() const { return flr_; } in function_library_runtime()
65 FunctionLibraryRuntime* flr_; member in tensorflow::__anone5cc3cd90111::TestEnv
|
| D | kernel_and_device.cc | 70 if (flr_ == nullptr) { in Init()
75 TF_RETURN_IF_ERROR(flr_->CreateKernel(ndef, &k)); in Init()
84 if (flr_ == nullptr) { in Init()
92 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in Init()
239 params.function_library = flr_; in Run()
|
| D | kernel_and_device.h | 72 flr_(flr), in KernelAndDevice()
123 FunctionLibraryRuntime* const flr_; // can be null variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 93 allocator_, flr_, node_def, in CreateOpKernel()
113 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
161 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
187 params_->function_library = flr_; in CreateOpKernelContext()
|
| D | dataset_test_base.h | 188 FunctionLibraryRuntime* flr_; // Owned by `pflr_`. variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 151 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
166 FunctionLibraryRuntime* flr_;
|
| D | tf2xla_util.cc | 698 TF_RETURN_IF_ERROR(flr_->Instantiate(func_name, attrs, handle)); in GetOrInstantiate()
706 result.Update(flr_->ReleaseHandle(iter.second)); in ReleaseAllHandles()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 298 return flr_; in mutable_function_library_runtime()
410 if (!flr_) { in InitializeFunctionLibraryRuntime()
420 flr_ = process_flr_->GetFLR(device_mgr_->ListDevices()[0]->name()); in InitializeFunctionLibraryRuntime()
433 FunctionLibraryRuntime* flr_ = nullptr; member in tensorflow::grappler::__anon35816ea30111::FunctionOptimizerContext
|