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