| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 61 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in IteratorResource() argument
66 std::move(flib_def), std::move(pflr), lib, nullptr /* iterator */)), in IteratorResource()
149 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Restore() local
150 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in Restore()
160 std::move(flib_def), std::move(pflr), lib, nullptr /* iterator */); in Restore()
208 iterator_state_->flib_def, iterator_state_->pflr, in SetIteratorFromDataset()
227 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in SetIteratorFromDataset() local
229 ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in SetIteratorFromDataset()
231 new_state->pflr = std::move(pflr); in SetIteratorFromDataset()
268 std::shared_ptr<ProcessFunctionLibraryRuntime> pflr, in State()
[all …]
|
| D | multi_device_iterator_ops.cc | 50 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr, in MultiDeviceIterator() argument
58 pflr_(std::move(pflr)), in MultiDeviceIterator()
407 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr(nullptr); in Compute() local
409 &flib_def, &pflr, &lib)); in Compute()
423 std::move(flib_def), std::move(pflr), lib, in Compute()
435 [this, context, lib, &flib_def, &pflr, in Compute()
441 std::move(flib_def), std::move(pflr), in Compute()
|
| D | iterator_ops.h | 67 std::unique_ptr<ProcessFunctionLibraryRuntime>* pflr);
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.cc | 57 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr = in FromGraph() local
68 TF_RETURN_IF_ERROR(graph_runner.Run(&graph, pflr->GetFLR("/device:CPU:0"), in FromGraph()
79 WrapUnique(new Dataset(dataset, device_mgr.release(), pflr.release(), in FromGraph()
112 ProcessFunctionLibraryRuntime* pflr, in Dataset() argument
117 pflr_(pflr), in Dataset()
|
| D | standalone.h | 106 ProcessFunctionLibraryRuntime* pflr,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 238 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run() local
243 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in Run()
|
| D | tf2xla_util_test.cc | 292 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in TEST() local
297 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 194 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc() argument
201 pflr_(pflr), in KernelAndDeviceFunc()
|
| D | kernel_and_device_test.cc | 59 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr() function in tensorflow::__anone5cc3cd90111::TestEnv
|
| D | context.h | 96 ProcessFunctionLibraryRuntime* pflr() const { return pflr_.get(); } in pflr() function
|
| D | execute.cc | 445 flr, ctx->pflr(), std::move(input_dev_ptrs), runner, in EagerLocalExecute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 286 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in PartiallyDeclusterGraph() local
289 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in PartiallyDeclusterGraph()
|
| D | mark_for_compilation_pass.cc | 453 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in FindCompilationCandidates() local
457 pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice); in FindCompilationCandidates()
|
| D | encapsulate_subgraphs_pass.cc | 2563 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in Run() local
2568 pflr->GetFLR("/job:localhost/replica:0/task:0/device:CPU:0"); in Run()
|
| D | encapsulate_subgraphs_pass_test.cc | 511 auto pflr = absl::make_unique<ProcessFunctionLibraryRuntime>( in Encapsulate() local
514 auto flr = pflr->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in Encapsulate()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 129 std::unique_ptr<ProcessFunctionLibraryRuntime> pflr( in OptimizeGraph() local
133 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 2158 tensorflow::ProcessFunctionLibraryRuntime pflr( in InlineAllFunctions() local
2162 flr = pflr.GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InlineAllFunctions()
|