| /external/tensorflow/tensorflow/compiler/jit/ |
| D | create_xla_launch_op.cc | 68 Status CompilationRequested(const FunctionLibraryRuntime& flr, in CompilationRequested() argument
71 flr.GetFunctionLibraryDefinition()->Find(node_def.name()); in CompilationRequested()
92 Status status = flr.GetFunctionLibraryDefinition()->GetAttr( in CompilationRequested()
114 Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr, in GetBodyAndConstantsAndResources() argument
123 flr->Instantiate(node_def.op(), AttrSlice(&node_def.attr()), &handle)); in GetBodyAndConstantsAndResources()
124 *fbody = flr->GetFunctionBody(handle); in GetBodyAndConstantsAndResources()
131 /*compile_time_const_nodes=*/nullptr, flr)); in GetBodyAndConstantsAndResources()
153 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def, in CreateXlaLaunchOp() argument
155 TF_RETURN_IF_ERROR(CompilationRequested(*flr, node_def)); in CreateXlaLaunchOp()
161 if (!IsCompilable(flr, node_def)) { in CreateXlaLaunchOp()
[all …]
|
| D | extract_outside_compilation_pass.h | 92 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr,
104 FunctionLibraryRuntime* flr, FunctionLibraryDefinition* fld);
|
| D | extract_outside_compilation_pass.cc | 1279 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions() argument
1314 func, new_func_name, host_func_name, host_compute_core, flr, fld, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1371 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1376 host_compute_core, flr, fld, shape_inference_graphs, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1436 cond, cond_xla_func_name, cond_host_func_name, host_compute_core, flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1440 body, body_xla_func_name, body_host_func_name, host_compute_core, flr, in ExtractOutsideCompilationForNodesWithAssociatedFunctions()
1589 const std::map<string, int>& host_compute_core, FunctionLibraryRuntime* flr, in ExtractOutsideCompilationForFunction() argument
1596 flr->Instantiate(func_name, AttrSlice(&func_name_attrs.attr()), &handle)); in ExtractOutsideCompilationForFunction()
1599 auto s = flr->ReleaseHandle(handle); in ExtractOutsideCompilationForFunction()
1604 const FunctionBody* fbody = flr->GetFunctionBody(handle); in ExtractOutsideCompilationForFunction()
[all …]
|
| D | create_xla_launch_op.h | 30 Status CreateXlaLaunchOp(FunctionLibraryRuntime* flr, const NodeDef& node_def,
|
| D | mark_for_compilation_pass.h | 54 bool IsCompilable(FunctionLibraryRuntime* flr, const NodeDef& ndef);
|
| D | mark_for_compilation_pass.cc | 653 bool IsCompilable(FunctionLibraryRuntime* flr, const NodeDef& ndef) { in IsCompilable() argument
654 Device* device = flr->device(); in IsCompilable()
669 return IsCompilableCall(ndef, jit_device_type, op_filter, 0, flr); in IsCompilable()
|
| D | encapsulate_subgraphs_pass.cc | 2567 FunctionLibraryRuntime* flr = in Run() local
2569 if (flr == nullptr) { in Run()
2576 [flr](const std::vector<OutputTensor>& arg_source_tensors, in Run()
2607 OptimizeGraph(flr, subgraph, graph_optimizer_options); in Run()
2613 /*compile_time_const_nodes=*/nullptr, flr)); in Run()
|
| D | extract_outside_compilation_pass_test.cc | 251 auto flr = pflr_->GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in ExtractOutsideCompilationTest() local
255 flr, fld, shape_inference_graphs, has_outside_compilation); in ExtractOutsideCompilationTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_threadpool_test.cc | 72 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run() argument
90 flr->Run(opts, handle, args, &out, [&status, &done](const Status& s) { in Run()
110 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
113 return flr->Instantiate(name, attrs, handle); in Instantiate()
116 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
120 return flr->Instantiate(name, attrs, options, handle); in Instantiate()
123 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
127 return InstantiateAndRun(flr, name, attrs, in InstantiateAndRun()
133 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
139 Status status = flr->Instantiate(name, attrs, options, &handle); in InstantiateAndRun()
[all …]
|
| D | process_function_library_runtime.cc | 148 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceIncarnation() local
149 if (flr == nullptr) { in GetDeviceIncarnation()
152 *incarnation = flr->device()->attributes().incarnation(); in GetDeviceIncarnation()
159 FunctionLibraryRuntime* flr = GetFLR(device_name); in GetDeviceContext() local
160 if (flr == nullptr) { in GetDeviceContext()
163 Device* device = flr->device(); in GetDeviceContext()
170 auto* dev_info = flr->device()->tensorflow_gpu_device_info(); in GetDeviceContext()
602 FunctionLibraryRuntime* flr = GetFLR(options.target); in InstantiateMultiDevice() local
603 if (flr == nullptr) { in InstantiateMultiDevice()
608 default_device = flr->device(); in InstantiateMultiDevice()
[all …]
|
| D | process_function_library_runtime_test.cc | 229 FunctionLibraryRuntime* flr = in TEST_F() local
231 EXPECT_NE(flr, nullptr); in TEST_F()
236 FunctionLibraryRuntime* flr = in TEST_F() local
238 EXPECT_NE(flr, nullptr); in TEST_F()
239 EXPECT_EQ(flr->device(), device0_); in TEST_F()
240 flr = proc_flr_->GetFLR("/job:a/replica:0/task:0/device:CPU:0"); in TEST_F()
241 EXPECT_NE(flr, nullptr); in TEST_F()
242 EXPECT_EQ(flr->device(), device0_); in TEST_F()
243 flr = proc_flr_->GetFLR("/device:CPU:0"); in TEST_F()
244 EXPECT_NE(flr, nullptr); in TEST_F()
[all …]
|
| D | function_test.cc | 169 Status Run(FunctionLibraryRuntime* flr, FunctionLibraryRuntime::Handle handle, in Run() argument
187 flr->Run(opts, handle, args, &out, [&status, &done](const Status& s) { in Run()
207 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
210 return flr->Instantiate(name, attrs, handle); in Instantiate()
213 Status Instantiate(FunctionLibraryRuntime* flr, const string& name, in Instantiate() argument
217 return flr->Instantiate(name, attrs, options, handle); in Instantiate()
220 Status InstantiateAndRun(FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
224 return InstantiateAndRun(flr, name, attrs, in InstantiateAndRun()
230 FunctionLibraryRuntime* flr, const string& name, in InstantiateAndRun() argument
236 Status status = flr->Instantiate(name, attrs, options, &handle); in InstantiateAndRun()
[all …]
|
| D | function.h | 275 FunctionLibraryRuntime& flr,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 66 FunctionLibraryRuntime* flr, in KernelAndDevice() argument
70 : device_(flr == nullptr ? nullptr : flr->device()), in KernelAndDevice()
72 flr_(flr), in KernelAndDevice()
134 FunctionLibraryRuntime* flr, in KernelAndDeviceOp() argument
138 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceOp()
194 FunctionLibraryRuntime* flr, ProcessFunctionLibraryRuntime* pflr, in KernelAndDeviceFunc() argument
199 : KernelAndDevice(flr, runner, std::move(collective_executor), in KernelAndDeviceFunc()
|
| D | execute.cc | 419 FunctionLibraryRuntime* flr = in EagerLocalExecute() local
421 if (device != nullptr && flr == nullptr) { in EagerLocalExecute()
426 auto runner = (flr != nullptr && flr->runner() != nullptr) ? flr->runner() in EagerLocalExecute()
445 flr, ctx->pflr(), std::move(input_dev_ptrs), runner, in EagerLocalExecute()
452 ctx->GetRendezvous(), ctx->LogMemory(), flr, runner, in EagerLocalExecute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 98 FunctionLibraryDefinition* fld, FunctionLibraryRuntime* flr, in FunctionalizeControlFlowForFunction() argument
105 TF_RETURN_IF_ERROR(flr->Instantiate(func_name, AttrSlice(&attrs), &handle)); in FunctionalizeControlFlowForFunction()
108 auto s = flr->ReleaseHandle(handle); in FunctionalizeControlFlowForFunction()
113 const FunctionBody* body = flr->GetFunctionBody(handle); in FunctionalizeControlFlowForFunction()
169 name, new_name, associated_function.attrs(), fld, flr, in FunctionalizeControlFlowForFunction()
242 FunctionLibraryRuntime* flr = in Run() local
270 func.name(), new_func_name, func.attr(), options.flib_def, flr, in Run()
|
| D | tf2xla_util.h | 151 CachedFunctionHandles(FunctionLibraryRuntime* flr) : flr_(flr) {} in CachedFunctionHandles() argument
|
| D | tf2xla_util_test.cc | 296 FunctionLibraryRuntime* flr = in TEST() local
299 CachedFunctionHandles cached_function_handles(flr); in TEST()
311 const FunctionBody* body = flr->GetFunctionBody(first_handle); in TEST()
|
| /external/u-boot/drivers/i2c/ |
| D | at91_i2c.h | 50 u32 flr; member
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 133 FunctionLibraryRuntime* flr = pflr->GetFLR(cpu_device->name()); in OptimizeGraph() local
146 optimizer.Optimize(flr, env, cpu_device, &graphptr, /*shape_map=*/nullptr); in OptimizeGraph()
|
| /external/mesa3d/src/mesa/swrast/ |
| D | s_texfilter.c | 212 const GLint flr = IFLOOR(s); in linear_texel_locations() local
213 if (flr & 1) in linear_texel_locations()
214 u = 1.0F - (s - (GLfloat) flr); in linear_texel_locations()
216 u = s - (GLfloat) flr; in linear_texel_locations()
340 const GLint flr = IFLOOR(s); in nearest_texel_location() local
342 if (flr & 1) in nearest_texel_location()
343 u = 1.0F - (s - (GLfloat) flr); in nearest_texel_location()
345 u = s - (GLfloat) flr; in nearest_texel_location()
|
| /external/deqp-deps/glslang/glslang/MachineIndependent/ |
| D | Constant.cpp | 647 double flr = floor(unionArray[i].getDConst()); in fold() local
648 bool even = flr / 2.0 == floor(flr / 2.0); in fold()
|
| /external/mesa3d/src/gallium/drivers/softpipe/ |
| D | sp_tex_sample.c | 202 int flr; in wrap_nearest_mirror_repeat() local
206 flr = util_ifloor(s); in wrap_nearest_mirror_repeat()
208 if (flr & 1) in wrap_nearest_mirror_repeat()
334 int flr; in wrap_linear_mirror_repeat() local
338 flr = util_ifloor(s); in wrap_linear_mirror_repeat()
340 if (flr & 1) in wrap_linear_mirror_repeat()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 1159 FunctionLibraryRuntime* flr = ctx->mutable_function_library_runtime(); in InlineSymbolicGradient() local
1162 const bool expanded = ExpandInlineFunctions(flr, &graph); in InlineSymbolicGradient()
1184 while (ExpandInlineFunctions(flr, &graph)) { in InlineSymbolicGradient()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 2161 tensorflow::FunctionLibraryRuntime* flr; in InlineAllFunctions() local
2162 flr = pflr.GetFLR("/job:localhost/replica:0/task:0/cpu:0"); in InlineAllFunctions()
2177 while (tensorflow::ExpandInlineFunctions(flr, &graph)) { in InlineAllFunctions()
|