| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator.cc | 39 static bool IsCompilable(FunctionLibraryRuntime* flr, const NodeDef& ndef, in IsCompilable() argument
42 Device* device = flr->device(); in IsCompilable()
64 return checker.IsCompilableCall(ndef, flr); in IsCompilable()
68 checker.FindUncompilableNodes(ndef, flr); in IsCompilable()
74 const FunctionLibraryRuntime& flr, in CanCreateKernel() argument
77 !XlaOpRegistry::IsCompilationDevice(flr.device()->device_type()); in CanCreateKernel()
80 static Status CreateXlaKernel(FunctionLibraryRuntime* flr, in CreateXlaKernel() argument
99 flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); in CreateXlaKernel()
103 if (flr->config_proto()) { in CreateXlaKernel()
104 config_proto = *flr->config_proto(); in CreateXlaKernel()
[all …]
|
| D | xla_kernel_creator.h | 33 const FunctionLibraryRuntime& flr,
37 Status CreateKernel(FunctionLibraryRuntime* flr,
|
| D | force_xla_constants_on_host_pass.cc | 32 FunctionLibraryRuntime* flr = in Run() local
46 flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); in Run()
|
| D | extract_outside_compilation_pass.h | 94 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr,
106 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld,
|
| D | get_compiler_ir.cc | 56 FunctionLibraryRuntime* flr = pflr->GetFLR(dev->name()); in GetCompilerIr() local
63 flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); in GetCompilerIr()
106 GenerateCompilerOptions(*cache, *flr, dev, in GetCompilerIr()
|
| D | extract_outside_compilation_pass.cc | 1809 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForFuncCallNode() argument
1839 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForFuncCallNode()
1895 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForIfNode() argument
1916 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
1921 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForIfNode()
2014 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld, in ExtractOutsideCompilationForWhileNode() argument
2032 cond, cond_xla_func_name, cond_host_func_name, host_compute_core, flr, in ExtractOutsideCompilationForWhileNode()
2036 body, body_xla_func_name, body_host_func_name, host_compute_core, flr, in ExtractOutsideCompilationForWhileNode()
2114 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions() argument
2132 host_compute_core, g, n, flr, fld, host_graphs, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_threadpool_test.cc | 77 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run() argument
95 flr->Run(opts, handle, args, &out, [&status, &done](const Status& s) { in Run()
115 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
118 return flr->Instantiate(name, attrs, handle); in Instantiate()
121 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
125 return flr->Instantiate(name, attrs, options, handle); in Instantiate()
128 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
132 return InstantiateAndRun(flr, name, attrs, in InstantiateAndRun()
138 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
144 Status status = flr->Instantiate(name, attrs, options, &handle); in InstantiateAndRun()
[all …]
|
| D | process_function_library_runtime.cc | 156 FunctionLibraryRuntime* flr = nullptr; in GetRetTypes() local
166 flr = GetFLR(fiter->second->target_device()); in GetRetTypes()
169 if (flr != nullptr) { in GetRetTypes()
170 return flr->GetRetTypes(h, ret_types); in GetRetTypes()
177 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation() local
178 if (flr == nullptr) { in GetDeviceIncarnation()
181 *incarnation = flr->device()->attributes().incarnation(); in GetDeviceIncarnation()
188 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext() local
189 if (flr == nullptr) { in GetDeviceContext()
192 Device* device = flr->device(); in GetDeviceContext()
[all …]
|
| D | process_function_library_runtime_test.cc | 343 FunctionLibraryRuntime* flr = in TEST_F() local
345 EXPECT_NE(flr, nullptr); in TEST_F()
350 FunctionLibraryRuntime* flr = in TEST_F() local
352 EXPECT_NE(flr, nullptr); in TEST_F()
353 EXPECT_EQ(flr->device(), device0_); in TEST_F()
354 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0"); in TEST_F()
355 EXPECT_NE(flr, nullptr); in TEST_F()
356 EXPECT_EQ(flr->device(), device0_); in TEST_F()
357 flr = proc_flr_->GetFLR("/device:CPU:0"); in TEST_F()
358 EXPECT_NE(flr, nullptr); in TEST_F()
[all …]
|
| D | function_test.cc | 177 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run() argument
195 flr->Run(opts, handle, args, &out, [&status, &done](const Status& s) { in Run()
215 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
218 return flr->Instantiate(name, attrs, handle); in Instantiate()
221 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
225 return flr->Instantiate(name, attrs, options, handle); in Instantiate()
228 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
232 return InstantiateAndRun(flr, name, attrs, in InstantiateAndRun()
238 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
244 Status status = flr->Instantiate(name, attrs, options, &handle); in InstantiateAndRun()
[all …]
|
| D | function.cc | 432 FunctionLibraryRuntime* flr, OpKernel** kernel);
605 FunctionLibraryRuntime* flr, OpKernel** kernel) { in CreateKernel() argument
623 flr->GetFunctionLibraryDefinition(); in CreateKernel()
626 return CreateNonCachedKernel(device_, flr, props, graph_def_version_, in CreateKernel()
663 device_type, device_, device_->GetAllocator(AllocatorAttributes()), flr, in CreateKernel()
908 FunctionLibraryRuntime* flr; in CreateItem() local
913 flr = (*item)->overlay_flr in CreateItem()
919 flr->GetFunctionLibraryDefinition(); in CreateItem()
932 params.function_library = flr; in CreateItem()
933 if (flr == this) { in CreateItem()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 134 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr,
142 FunctionLibraryRuntime* flr, bool* any_function_modified, in FunctionalizeControlFlowForNodeAssociatedFunctions() argument
188 func_name, new_func_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForNodeAssociatedFunctions()
207 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction() argument
213 TF_RETURN_IF_ERROR(flr->Instantiate(func_name, AttrSlice(&attrs), &handle)); in FunctionalizeControlFlowForFunction()
216 auto s = flr->ReleaseHandle(handle); in FunctionalizeControlFlowForFunction()
221 const FunctionBody* body = flr->GetFunctionBody(handle); in FunctionalizeControlFlowForFunction()
237 func_map, g, fld, flr, function_modified, node_filter)); in FunctionalizeControlFlowForFunction()
289 FunctionLibraryRuntime* flr = in FunctionalizeControlFlow() local
295 &func_map, graph, library, flr, &modified, node_filter)); in FunctionalizeControlFlow()
[all …]
|
| D | tf2xla_util.h | 152 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles() argument
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 113 FunctionLibraryRuntime* flr, in KernelAndDevice() argument
117 : device_(flr == nullptr ? nullptr : flr->device()), in KernelAndDevice()
119 flr_(flr), in KernelAndDevice()
194 FunctionLibraryRuntime* flr, in KernelAndDeviceOp() argument
198 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
261 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc() argument
272 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 85 FunctionLibraryRuntime* flr) in IteratorResource() argument
89 std::move(pflr), flr, in IteratorResource()
96 collect_metrics_(flr->device()->device_type() == DEVICE_CPU) { in IteratorResource()
119 params.flr = captured_state->flr(); in GetNext()
202 iterator_state_->pflr(), iterator_state_->flr(), in Restore()
207 params.flr = new_state->flr(); in Restore()
236 iterator_state_->pflr(), iterator_state_->flr(), in SetIteratorFromDataset()
242 params.flr = new_state->flr(); in SetIteratorFromDataset()
474 FunctionLibraryRuntime* flr; in Compute() local
483 flr = CreatePrivateFLR(context, &device_mgr, &flib_def, &pflr); in Compute()
[all …]
|
| D | dataset_ops.cc | 136 FunctionLibraryRuntime* flr; in Compute() local
140 ctx->function_library()->Clone(&flib_def, &pflr, &flr, true)); in Compute()
151 GraphRunner graph_runner(flr->device()); in Compute()
153 graph_runner.Run(&graph, flr, {}, {output_node}, &outputs)); in Compute()
|
| D | multi_device_iterator_ops.cc | 60 FunctionLibraryRuntime* flr, in MultiDeviceIterator() argument
67 flr_(flr), in MultiDeviceIterator()
106 params.flr = flr_; in GetNextFromShard()
136 FunctionLibraryRuntime* const flr() { in flr() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
428 FunctionLibraryRuntime* flr; in Compute() local
432 &flib_def, &pflr, &flr)); in Compute()
434 absl::make_unique<FunctionHandleCache>(flr); in Compute()
446 std::move(flib_def), std::move(pflr), flr, in Compute()
458 [this, context, flr, &flib_def, &pflr, in Compute()
465 flr, std::move(function_handle_cache)); in Compute()
[all …]
|
| D | iterator_ops.h | 40 FunctionLibraryRuntime* flr);
80 FunctionLibraryRuntime* flr, in State() argument
83 flr_(flr), in State()
85 function_handle_cache_(absl::make_unique<FunctionHandleCache>(flr)), in State()
98 FunctionLibraryRuntime* flr() { return flr_; } in flr() function
|
| D | rewrite_utils.cc | 163 FunctionLibraryRuntime* flr = nullptr; in RewriteDataset() local
167 ctx->function_library()->Clone(&lib_def, &pflr, &flr, true)); in RewriteDataset()
176 GraphRunner graph_runner(flr->device()); in RewriteDataset()
179 graph_runner.Run(&graph, flr, input_list, {output_node}, &outputs)); in RewriteDataset()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 458 FunctionLibraryRuntime* flr, GraphShapeInfo* shape_info) { in RunShapeInferenceOnComputation() argument
483 flr != nullptr ? flr->GetFunctionLibraryDefinition() : nullptr, in RunShapeInferenceOnComputation()
490 std::unique_ptr<Graph>* graph, FunctionLibraryRuntime* flr, in OptimizeGraph() argument
515 metadata, arg_shapes, graph->get(), flr, &shape_info)); in OptimizeGraph()
521 optimizer.Optimize(flr, flr->env(), flr->device(), graph, optimizer_opts); in OptimizeGraph()
528 metadata, arg_shapes, graph->get(), flr, &shape_info)); in OptimizeGraph()
531 optimizer.Optimize(flr, flr->env(), flr->device(), graph, &shape_map); in OptimizeGraph()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 278 FunctionLibraryRuntime* flr,
310 const Node* replicate_node, FunctionLibraryRuntime* flr,
529 int num_replicas, const Node& replicate_node, FunctionLibraryRuntime* flr,
553 FunctionLibraryRuntime* flr, Node* host_compute_key_placeholder_node,
567 FunctionLibraryRuntime* flr);
|
| D | host_training_loop_optimization_util.cc | 392 FunctionLibraryRuntime* flr, in DetectHostTrainingLoop() argument
410 TF_RETURN_IF_ERROR(flr->Instantiate(function.func_name(), in DetectHostTrainingLoop()
413 auto s = flr->ReleaseHandle(handle); in DetectHostTrainingLoop()
418 const FunctionBody* body = flr->GetFunctionBody(handle); in DetectHostTrainingLoop()
421 &function.func_name(), &function.attrs(), library, function_graph, flr, in DetectHostTrainingLoop()
|
| D | encapsulate_tpu_computations_pass.cc | 1876 Status LiftOutsideCompilationOnlyArgs(Graph* g, FunctionLibraryRuntime* flr,
1881 const FunctionBody& fbody, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef() argument
1886 fbody.graph, flr, fld, lifted_arg_count, rewritten)); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
2266 Graph* g, Node* call_node, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgsFromCallNode() argument
2285 flr->Instantiate(func.name(), AttrSlice(&func.attr()), &handle)); in LiftOutsideCompilationOnlyArgsFromCallNode()
2287 [&flr, &handle]() { flr->ReleaseHandle(handle).IgnoreError(); }); in LiftOutsideCompilationOnlyArgsFromCallNode()
2288 const FunctionBody* fbody = flr->GetFunctionBody(handle); in LiftOutsideCompilationOnlyArgsFromCallNode()
2352 Status LiftOutsideCompilationOnlyArgs(Graph* g, FunctionLibraryRuntime* flr, in LiftOutsideCompilationOnlyArgs() argument
2369 *body_fbody, flr, fld, lifted_arg_count, in LiftOutsideCompilationOnlyArgs()
2380 *then_branch_fbody, flr, fld, lifted_arg_count, in LiftOutsideCompilationOnlyArgs()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 139 auto flr = pflr_->GetFLR(device_->name()); in InitOpWithFunctionLibrary() local
142 node_def_, flr->GetFunctionLibraryDefinition(), &props); in InitOpWithFunctionLibrary()
144 status.Update(CreateOpKernel(device_type_, device_, allocator(), flr, in InitOpWithFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 48 FunctionLibraryRuntime* flr = ctx.func_lib(device); in CreateUncachedKernelAndDeviceOp() local
49 if (flr == nullptr) { in CreateUncachedKernelAndDeviceOp()
55 auto runner = (flr->runner() != nullptr) ? flr->runner() : ctx.runner(); in CreateUncachedKernelAndDeviceOp()
56 kernel->reset(new KernelAndDeviceOp(ctx.GetRendezvous(), ctx.LogMemory(), flr, in CreateUncachedKernelAndDeviceOp()
|