| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 81 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Init()
84 FunctionLibraryRuntime* flr_; member in tensorflow::XlaKernelCreatorTest
110 Status status = xla_kernel_creator.CreateKernel(flr_, callsite, &kernel_); in TEST_F()
133 xla_kernel_creator.CreateKernel(flr_, ToNodeProperties(R"proto( in TEST_F()
150 xla_kernel_creator.CreateKernel(flr_, ToNodeProperties(R"proto( in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 56 flr_ = pflr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0");
57 CHECK(flr_ != nullptr);
60 FunctionLibraryRuntime* function_library_runtime() const { return flr_; } in function_library_runtime()
67 FunctionLibraryRuntime* flr_; member in tensorflow::__anon9d2e75500111::TestEnv
|
| D | kernel_and_device.cc | 104 if (flr_ == nullptr) { in Init()
111 ndef, flr_->GetFunctionLibraryDefinition(), &props)); in Init()
112 TF_RETURN_IF_ERROR(flr_->CreateKernel(props, &k)); in Init()
138 if (flr_ == nullptr) { in InstantiateFunc()
146 function_def = flr_->GetFunctionLibraryDefinition()->Find(ndef.op()); in InstantiateFunc()
256 params.function_library = flr_; in Run()
|
| D | kernel_and_device.h | 119 flr_(flr), in KernelAndDevice()
182 FunctionLibraryRuntime* const flr_; // can be null variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 152 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles()
167 FunctionLibraryRuntime* flr_;
|
| D | tf2xla_util.cc | 697 TF_RETURN_IF_ERROR(flr_->Instantiate(func_name, attrs, handle)); in GetOrInstantiate()
705 result.Update(flr_->ReleaseHandle(iter.second)); in ReleaseAllHandles()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 83 flr_(flr), in State()
98 FunctionLibraryRuntime* flr() { return flr_; } in flr()
116 FunctionLibraryRuntime* flr_ = nullptr; // not owned variable
|
| D | multi_device_iterator_ops.cc | 67 flr_(flr), in MultiDeviceIterator()
70 DCHECK(flr_ != nullptr); in MultiDeviceIterator()
106 params.flr = flr_; in GetNextFromShard()
138 return flr_; in flr()
381 FunctionLibraryRuntime* const flr_ = nullptr; // not owned. member in tensorflow::data::__anond56f2ce10111::MultiDeviceIterator
|
| D | dataset_test_base.cc | 310 node_def, flr_->GetFunctionLibraryDefinition(), &props)); in CreateOpKernel()
312 device_type_, device_.get(), allocator_, flr_, in CreateOpKernel()
355 function_handle_cache_ = absl::make_unique<FunctionHandleCache>(flr_); in CreateIteratorContext()
412 flr_ = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InitFunctionLibraryRuntime()
449 params.function_library = flr_; in RunFunction()
454 return CreateNonCachedKernel(device_.get(), this->flr_, props, version, in RunFunction()
498 params->function_library = flr_; in CreateOpKernelContext()
|
| D | dataset_test_base.h | 764 FunctionLibraryRuntime* flr_; // Owned by `pflr_`. variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 98 flr_ = pflr_->GetFLR(device_->name()); in Benchmark()
102 params.function_library = flr_; in Benchmark()
106 return CreateNonCachedKernel(device_, flr_, props, graph_def_version, in Benchmark()
|
| D | kernel_benchmark_testlib.h | 96 FunctionLibraryRuntime* flr_; // Not owned. variable
|