| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 40 "kernels/activations.cc",
41 "kernels/add.cc",
42 "kernels/add_n.cc",
43 "kernels/arg_min_max.cc",
44 "kernels/basic_rnn.cc",
45 "kernels/batch_matmul.cc",
46 "kernels/batch_to_space_nd.cc",
47 "kernels/bidirectional_sequence_lstm.cc",
48 "kernels/bidirectional_sequence_rnn.cc",
49 "kernels/broadcast_to.cc",
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/tools/make/ |
| D | Makefile | 259 tensorflow/lite/micro/kernels/activations_test.cc \
260 tensorflow/lite/micro/kernels/add_test.cc \
261 tensorflow/lite/micro/kernels/arg_min_max_test.cc \
262 tensorflow/lite/micro/kernels/batch_to_space_nd_test.cc \
263 tensorflow/lite/micro/kernels/cast_test.cc \
264 tensorflow/lite/micro/kernels/ceil_test.cc \
265 tensorflow/lite/micro/kernels/circular_buffer_test.cc \
266 tensorflow/lite/micro/kernels/comparisons_test.cc \
267 tensorflow/lite/micro/kernels/concatenation_test.cc \
268 tensorflow/lite/micro/kernels/conv_test.cc \
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 26 "//tensorflow/core/kernels:ops_util",
35 "//tensorflow/core/kernels:concat_lib",
36 "//tensorflow/core/kernels:conv_2d",
37 "//tensorflow/core/kernels:eigen_contraction_kernel",
38 "//tensorflow/core/kernels:fill_functor",
39 "//tensorflow/core/kernels:gather_functor",
40 "//tensorflow/core/kernels:transpose_functor",
54 "//tensorflow/core/kernels:ops_testutil",
55 "//tensorflow/core/kernels:ops_util",
65 "//tensorflow/core/kernels:batch_matmul_op",
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | BUILD | 2 # Contains experimental kernels for datasets and iterators.
22 "//tensorflow/core/kernels/data:name_utils",
33 "//tensorflow/core/kernels/data:dataset_utils",
34 "//tensorflow/core/kernels/data:name_utils",
46 "//tensorflow/core/kernels/data:dataset_test_base",
47 "//tensorflow/core/kernels/data:range_dataset_op",
48 "//tensorflow/core/kernels/data:take_dataset_op",
64 "//tensorflow/core/kernels/data:rewrite_utils",
79 "//tensorflow/core/kernels/data:dataset_test_base",
80 "//tensorflow/core/kernels/data:shard_dataset_op",
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | BUILD | 1 # Description: Op kernels for sparse matrix operations.
28 name = "kernels",
35 "kernels.cc",
49 "kernels.h",
55 "kernels.h",
73 "//tensorflow/core/kernels:concat_lib",
74 "//tensorflow/core/kernels:constant_op",
75 "//tensorflow/core/kernels:cwise_op",
76 "//tensorflow/core/kernels:dense_update_functor",
77 "//tensorflow/core/kernels:fill_functor",
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | BUILD | 22 "//tensorflow/lite/kernels/internal:compatibility",
34 "//tensorflow/lite/kernels/internal:compatibility",
55 "//tensorflow/lite/kernels:kernel_util",
56 "//tensorflow/lite/kernels:op_macros",
57 "//tensorflow/lite/kernels/internal:optimized_base",
58 "//tensorflow/lite/kernels/internal:tensor",
71 "//tensorflow/lite/kernels:builtin_ops",
72 "//tensorflow/lite/kernels:test_util",
83 "//tensorflow/lite/kernels:cpu_backend_context",
84 "//tensorflow/lite/kernels/internal:optimized_base",
[all …]
|
| /external/skqp/src/compute/skc/platforms/cl_12/ |
| D | raster_builder_cl_12.c | 179 } kernels; member
311 cl(ReleaseKernel(impl->kernels.fills_expand)); in skc_raster_builder_pfn_release()
312 cl(ReleaseKernel(impl->kernels.rasterize_all)); in skc_raster_builder_pfn_release()
315 cl(ReleaseKernel(impl->kernels.rasterize_lines)); in skc_raster_builder_pfn_release()
316 cl(ReleaseKernel(impl->kernels.rasterize_quads)); in skc_raster_builder_pfn_release()
317 cl(ReleaseKernel(impl->kernels.rasterize_cubics)); in skc_raster_builder_pfn_release()
320 cl(ReleaseKernel(impl->kernels.segment)); in skc_raster_builder_pfn_release()
321 cl(ReleaseKernel(impl->kernels.rasters_alloc)); in skc_raster_builder_pfn_release()
322 cl(ReleaseKernel(impl->kernels.prefix)); in skc_raster_builder_pfn_release()
581 cl(SetKernelArg(impl->kernels.segment,0,SKC_CL_ARG(cohort->keys.drw))); in skc_raster_cohort_sort_prefix()
[all …]
|
| D | surface_cl_12.c | 49 } kernels; member
161 cl(ReleaseKernel(impl->kernels.render)); in skc_surface_pfn_release()
251 cl(SetKernelArg(impl->kernels.render,0,SKC_CL_ARG(si->layers.drN))); in skc_surface_grid_pfn_execute()
252 cl(SetKernelArg(impl->kernels.render,1,SKC_CL_ARG(si->groups.drN))); in skc_surface_grid_pfn_execute()
253 cl(SetKernelArg(impl->kernels.render,2,SKC_CL_ARG(si->extras.drN))); in skc_surface_grid_pfn_execute()
255 cl(SetKernelArg(impl->kernels.render,3,SKC_CL_ARG(ci->keys.drw))); in skc_surface_grid_pfn_execute()
256 cl(SetKernelArg(impl->kernels.render,4,SKC_CL_ARG(atomics->keys))); in skc_surface_grid_pfn_execute()
257 cl(SetKernelArg(impl->kernels.render,5,SKC_CL_ARG(ci->offsets.drw))); in skc_surface_grid_pfn_execute()
258 cl(SetKernelArg(impl->kernels.render,6,SKC_CL_ARG(atomics->offsets))); in skc_surface_grid_pfn_execute()
261 cl(SetKernelArg(impl->kernels.render,7,SKC_CL_ARG(impl->runtime->block_pool.blocks.drw))); in skc_surface_grid_pfn_execute()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | BUILD | 22 "//tensorflow/lite/kernels:kernel_util",
23 "//tensorflow/lite/kernels:lstm_shared",
24 "//tensorflow/lite/kernels:op_macros",
25 "//tensorflow/lite/kernels/internal:kernel_utils",
26 "//tensorflow/lite/kernels/internal:optimized_base",
27 "//tensorflow/lite/kernels/internal:reference",
28 "//tensorflow/lite/kernels/internal:tensor_utils",
42 "//tensorflow/lite/kernels:kernel_util",
43 "//tensorflow/lite/kernels:lstm_shared",
44 "//tensorflow/lite/kernels:op_macros",
[all …]
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | BUILD | 19 "@org_tensorflow//tensorflow/lite/kernels:kernel_util",
20 "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
21 "@org_tensorflow//tensorflow/lite/kernels/internal:types",
34 "@org_tensorflow//tensorflow/lite/kernels:test_util",
35 "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
47 "@org_tensorflow//tensorflow/lite/kernels:builtin_ops",
62 "@org_tensorflow//tensorflow/lite/kernels:kernel_util",
63 "@org_tensorflow//tensorflow/lite/kernels/internal:tensor",
64 "@org_tensorflow//tensorflow/lite/kernels/internal:types",
77 "@org_tensorflow//tensorflow/lite/kernels:test_util",
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | BUILD | 18 "//tensorflow/lite/kernels:builtin_ops",
19 "//tensorflow/lite/kernels:test_util",
20 "//tensorflow/lite/kernels/internal:reference_base",
21 "//tensorflow/lite/kernels/internal:tensor",
22 "//tensorflow/lite/kernels/internal:types",
61 "//tensorflow/lite/kernels:kernel_util",
62 "//tensorflow/lite/kernels:reshape_test_common",
63 "//tensorflow/lite/kernels/internal:types",
83 "//tensorflow/lite/kernels:builtin_ops",
84 "//tensorflow/lite/kernels:kernel_util",
[all …]
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 17 # framework/: Support for adding new ops & kernels
19 # ":ops" - defines TensorFlow ops, but no implementations / kernels
28 # ":all_kernels" - The cpu-specific kernels, plus gpu kernels if
34 # kernels.
38 # (kernels) that can also be run outside the tensorflow runtime. This
40 # kernels, without bringing in libraries that can bloat code size (e.g.,
146 "//quality/webanswers/brain/tokenization/custom_tf_ops/kernels/...",
397 "//tensorflow/core/kernels:required",
558 "//tensorflow/c/kernels:bitcast_op_lib",
559 "//tensorflow/c/kernels:histogram_summary_op_lib",
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry_test.cc | 70 for (const auto& kernels : registered_kernels) { in TEST() local
71 if (kernels.op() == "DummyDuplicateOp") { in TEST()
72 EXPECT_EQ(kernels.constraint_size(), 1); in TEST()
73 EXPECT_EQ(kernels.constraint(0).name(), "T"); in TEST()
74 if (kernels.device_type() == "XLA_CPU_JIT") { in TEST()
75 EXPECT_EQ(kernels.constraint(0).allowed_values().list().type(0), in TEST()
78 EXPECT_EQ(kernels.constraint(0).allowed_values().list().type(0), in TEST()
112 for (const auto& kernels : registered_kernels) { in TEST() local
114 EXPECT_NE(kernels.op(), "DummyInfeasibleTypeConstraintOp"); in TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | BUILD | 29 "//tensorflow/lite/kernels:kernel_util",
30 "//tensorflow/lite/kernels:padding",
31 "//tensorflow/lite/kernels/internal:common",
32 "//tensorflow/lite/kernels/internal:compatibility",
33 "//tensorflow/lite/kernels/internal:tensor",
34 "//tensorflow/lite/kernels/internal:tensor_utils",
35 "//tensorflow/lite/kernels/internal:types",
51 "//tensorflow/lite/kernels:test_main",
52 "//tensorflow/lite/kernels:test_util",
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | BUILD | 42 "//tensorflow/core/kernels:cwise_op",
43 "//tensorflow/core/kernels:quantization_utils",
44 "//tensorflow/core/kernels:quantized_ops",
45 "//tensorflow/core/kernels:reduction_ops",
46 "//tensorflow/core/kernels:remote_fused_graph_execute_utils",
47 "//tensorflow/core/kernels:remote_fused_graph_ops",
48 "//tensorflow/core/kernels:reshape_op",
49 "//tensorflow/core/kernels:softmax_op",
79 "//tensorflow/core/kernels:remote_fused_graph_execute_utils",
113 "//tensorflow/core/kernels:remote_fused_graph_execute_utils",
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | BUILD | 97 "//tensorflow/core/kernels:cast_op",
98 "//tensorflow/core/kernels:fill_functor",
102 "//tensorflow/core/kernels:transpose_functor",
167 "//tensorflow/core/kernels:cwise_op",
175 "//tensorflow/core/kernels:cwise_op",
219 "//tensorflow/core/kernels:fill_functor",
228 "//tensorflow/core/kernels:transpose_functor",
254 "//tensorflow/core/kernels:cwise_op",
270 "//tensorflow/core/kernels:transpose_functor",
284 "//tensorflow/core/kernels:batch_matmul_op",
[all …]
|
| /external/llvm-project/mlir/test/Conversion/GPUToSPIRV/ |
| D | builtins.mlir | 6 gpu.launch_func @kernels::@builtin_workgroup_id_x
13 gpu.module @kernels {
30 gpu.launch_func @kernels::@builtin_workgroup_id_y
37 gpu.module @kernels {
54 gpu.launch_func @kernels::@builtin_workgroup_id_z
61 gpu.module @kernels {
78 gpu.launch_func @kernels::@builtin_workgroup_size_x
84 gpu.module @kernels {
103 gpu.launch_func @kernels::@builtin_workgroup_size_y
109 gpu.module @kernels {
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | BUILD | 47 "//tensorflow/lite/kernels/internal:compatibility",
48 "//tensorflow/lite/kernels/internal:tensor",
64 "//tensorflow/lite/kernels/internal:reference",
85 "//tensorflow/lite/kernels:kernel_util",
86 "//tensorflow/lite/kernels/internal:compatibility",
87 "//tensorflow/lite/kernels/internal:tensor",
108 "//tensorflow/lite/kernels:op_macros",
109 "//tensorflow/lite/kernels/internal:compatibility",
110 "//tensorflow/lite/micro/kernels:conv",
111 "//tensorflow/lite/micro/kernels:ethosu",
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 1675 "//tensorflow/core/kernels:required",
1733 "//tensorflow/core/kernels:function_ops",
1787 "//tensorflow/core/kernels:ops_util",
1863 "//tensorflow/core/kernels:ops_util",
1904 "//tensorflow/core/kernels:ops_util",
2053 "//tensorflow/core/kernels:cast_op",
2082 "//tensorflow/core/kernels:bcast_ops",
2083 "//tensorflow/core/kernels:cast_op",
2084 "//tensorflow/core/kernels:concat_op",
2085 "//tensorflow/core/kernels:cwise_op",
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/vexriscv/ |
| D | README.md | 10 The optimized kernels for
24 [README](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/micro/kernels/vexrisc…
29 To use VexRISC-V optimized kernels instead of reference kernel add
30 `TAGS=vexriscv` to the make command. The kernels that doesn't have optimization
31 for a certain micro architecture fallback to use TFLM reference kernels.
45 ## Optimized kernels
47 The following kernels are optimized specific to VexRISCV
49 …tps://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/micro/kernels/vexriscv/doc/Dept…
|
| /external/OpenCL-CTS/test_conformance/math_brute_force/ |
| D | mad_float.cpp | 117 cl_kernel *kernels; member
127 return BuildKernel(info->nameInCode, i, info->kernels + i, in BuildKernelFn()
138 cl_kernel kernels[VECTOR_SIZE_COUNT]; in TestFunc_mad_Float() local
147 BuildKernelInfo build_info = { gMinVectorSizeIndex, kernels, programs, in TestFunc_mad_Float()
210 if ((error = clSetKernelArg(kernels[j], 0, sizeof(gOutBuffer[j]), in TestFunc_mad_Float()
216 if ((error = clSetKernelArg(kernels[j], 1, sizeof(gInBuffer), in TestFunc_mad_Float()
222 if ((error = clSetKernelArg(kernels[j], 2, sizeof(gInBuffer2), in TestFunc_mad_Float()
228 if ((error = clSetKernelArg(kernels[j], 3, sizeof(gInBuffer3), in TestFunc_mad_Float()
236 clEnqueueNDRangeKernel(gQueue, kernels[j], 1, NULL, in TestFunc_mad_Float()
295 clReleaseKernel(kernels[k]); in TestFunc_mad_Float()
|
| D | mad_double.cpp | 119 cl_kernel *kernels; member
129 return BuildKernel(info->nameInCode, i, info->kernels + i, in BuildKernelFn()
137 cl_kernel kernels[VECTOR_SIZE_COUNT]; in TestFunc_mad_Double() local
148 BuildKernelInfo build_info = { gMinVectorSizeIndex, kernels, programs, in TestFunc_mad_Double()
211 if ((error = clSetKernelArg(kernels[j], 0, sizeof(gOutBuffer[j]), in TestFunc_mad_Double()
217 if ((error = clSetKernelArg(kernels[j], 1, sizeof(gInBuffer), in TestFunc_mad_Double()
223 if ((error = clSetKernelArg(kernels[j], 2, sizeof(gInBuffer2), in TestFunc_mad_Double()
229 if ((error = clSetKernelArg(kernels[j], 3, sizeof(gInBuffer3), in TestFunc_mad_Double()
237 clEnqueueNDRangeKernel(gQueue, kernels[j], 1, NULL, in TestFunc_mad_Double()
296 clReleaseKernel(kernels[k]); in TestFunc_mad_Double()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | BUILD | 86 "//tensorflow/core/tpu/kernels:tpu_compilation_cache_entry",
87 "//tensorflow/core/tpu/kernels:tpu_compilation_cache_interface",
88 "//tensorflow/core/tpu/kernels:tpu_compilation_cache_key",
89 "//tensorflow/core/tpu/kernels:tpu_compilation_cache_lookup",
90 "//tensorflow/core/tpu/kernels:tpu_compile_op_common",
91 "//tensorflow/core/tpu/kernels:tpu_compile_op_hdrs",
92 "//tensorflow/core/tpu/kernels:tpu_mesh_state_interface",
93 "//tensorflow/core/tpu/kernels:tpu_op_consts",
94 "//tensorflow/core/tpu/kernels:tpu_op_util",
95 "//tensorflow/core/tpu/kernels:tpu_program_group",
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/tools/ci_build/ |
| D | test_code_style.sh | 50 kernels/internal/reference/ \
53 -e kernels/internal/reference/integer_ops/ \
54 -e kernels/internal/reference/reference_ops.h \
78 kernels/internal/reference/ \
80 -e kernels/internal/reference/integer_ops/ \
81 -e kernels/internal/reference/reference_ops.h \
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | BUILD | 27 "//tensorflow/lite/kernels:kernel_util",
28 "//tensorflow/lite/kernels/internal:tensor",
45 "//tensorflow/lite/kernels:test_main",
46 "//tensorflow/lite/kernels:test_util",
47 "//tensorflow/lite/kernels/internal:tensor",
67 "//tensorflow/lite/kernels:builtin_ops",
|