| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 1 tensorflow/c/kernels/bitcast_op.cc
8 tensorflow/core/kernels/aggregate_ops.cc
9 tensorflow/core/kernels/argmax_op.cc
10 tensorflow/core/kernels/avgpooling_op.cc
11 tensorflow/core/kernels/batch_matmul_op_real.cc
12 tensorflow/core/kernels/batch_norm_op.cc
13 tensorflow/core/kernels/batchtospace_op.cc
14 tensorflow/core/kernels/bcast_ops.cc
15 tensorflow/core/kernels/bias_op.cc
16 tensorflow/core/kernels/boosted_trees/prediction_ops.cc
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | tflite_static.bp | 42 "kernels/elementwise.cc",
43 "kernels/split.cc",
44 "kernels/topk_v2.cc",
45 "kernels/activations.cc",
46 "kernels/add.cc",
47 "kernels/add_n.cc",
48 "kernels/arg_min_max.cc",
49 "kernels/basic_rnn.cc",
50 "kernels/batch_to_space_nd.cc",
51 "kernels/bidirectional_sequence_lstm.cc",
[all …]
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_kernels.cmake | 21 "${tensorflow_source_dir}/tensorflow/core/kernels/*.h"
22 "${tensorflow_source_dir}/tensorflow/core/kernels/*.cc"
25 # Build a minimal subset of kernels to be able to run a test program.
27 "${tensorflow_source_dir}/tensorflow/core/kernels/bounds_check.h"
28 "${tensorflow_source_dir}/tensorflow/core/kernels/constant_op.h"
29 "${tensorflow_source_dir}/tensorflow/core/kernels/constant_op.cc"
30 "${tensorflow_source_dir}/tensorflow/core/kernels/fill_functor.h"
31 "${tensorflow_source_dir}/tensorflow/core/kernels/fill_functor.cc"
32 "${tensorflow_source_dir}/tensorflow/core/kernels/matmul_op.h"
33 "${tensorflow_source_dir}/tensorflow/core/kernels/matmul_op.cc"
[all …]
|
| /external/tensorflow/tensorflow/contrib/ignite/ |
| D | BUILD | 32 "kernels/igfs/igfs.h",
54 "kernels/client/ignite_client.h",
55 "kernels/client/ignite_byte_swapper.h",
56 "kernels/client/ignite_plain_client.h",
57 "kernels/client/ignite_ssl_wrapper.h",
58 "kernels/client/ignite_ssl_wrapper.cc",
60 "kernels/client/ignite_plain_client_unix.cc",
62 "kernels/client/ignite_plain_client_windows.cc",
77 "kernels/dataset/ignite_binary_object_parser.cc",
78 "kernels/dataset/ignite_binary_object_parser.h",
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | BUILD | 31 "//tensorflow/lite/kernels:test_util",
47 "//tensorflow/lite/kernels/internal:tensor_utils",
67 "//tensorflow/lite/kernels/internal:optimized",
114 "//tensorflow/lite/kernels/internal:quantization_util",
115 "//tensorflow/lite/kernels/internal:round",
249 "//tensorflow/lite/kernels:gemm_support",
250 "//tensorflow/lite/kernels/internal:audio_utils",
251 "//tensorflow/lite/kernels/internal:kernel_utils",
252 "//tensorflow/lite/kernels/internal:optimized",
253 "//tensorflow/lite/kernels/internal:optimized_base",
[all …]
|
| /external/tensorflow/tensorflow/contrib/image/ |
| D | BUILD | 25 "kernels/bipartite_match_op.cc",
26 "kernels/image_ops.cc",
27 "kernels/image_ops.h",
28 "kernels/segmentation_ops.cc",
29 "kernels/segmentation_ops.h",
33 "kernels/image_ops_gpu.cu.cc",
34 "kernels/image_ops.h",
41 "kernels/bipartite_match_op.cc",
42 "kernels/image_ops.cc",
43 "kernels/image_ops.h",
[all …]
|
| /external/tensorflow/tensorflow/contrib/bigtable/ |
| D | BUILD | 27 kernels = [
44 "kernels/bigtable_kernels.cc",
45 "kernels/bigtable_lookup_dataset_op.cc",
46 "kernels/bigtable_prefix_key_dataset_op.cc",
47 "kernels/bigtable_range_key_dataset_op.cc",
48 "kernels/bigtable_sample_keys_dataset_op.cc",
49 "kernels/bigtable_sample_key_pairs_dataset_op.cc",
50 "kernels/bigtable_scan_dataset_op.cc",
89 # A library for use in the bigtable kernels.
92 srcs = ["kernels/bigtable_lib.cc"],
[all …]
|
| /external/tensorflow/tensorflow/contrib/coder/ |
| D | BUILD | 26 "kernels/range_coder.cc",
29 "kernels/range_coder.h",
41 srcs = ["kernels/range_coder_test.cc"],
60 srcs = ["kernels/range_coder_ops_util.cc"],
61 hdrs = ["kernels/range_coder_ops_util.h"],
72 "kernels/range_coder_ops.cc",
88 srcs = ["kernels/range_coder_ops_test.cc"],
100 "//tensorflow/core/kernels:ops_testutil",
106 srcs = ["kernels/pmf_to_cdf_op.cc"],
118 srcs = ["kernels/pmf_to_cdf_op_test.cc"],
[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/skia/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/contrib/rnn/ |
| D | BUILD | 48 kernels = [
173 "kernels/blas_gemm.cc",
174 "kernels/blas_gemm.h",
175 "kernels/lstm_ops.cc",
176 "kernels/lstm_ops.h",
180 "kernels/blas_gemm.h",
181 "kernels/lstm_ops_gpu.cu.cc",
182 "kernels/lstm_ops.h",
185 "//tensorflow/core/kernels:eigen_contraction_kernel",
186 "//tensorflow/core/kernels:eigen_helpers",
[all …]
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ |
| D | BUILD | 34 kernels = [
64 "kernels/fused_conv2d_bias_activation_op.cc",
65 "kernels/fused_conv2d_bias_activation_op.h",
66 "kernels/fused_conv_ops_gpu.h",
77 "//tensorflow/core/kernels:bounds_check",
78 "//tensorflow/core/kernels:conv_2d_hdrs",
79 "//tensorflow/core/kernels:conv_ops_gpu_hdrs",
80 "//tensorflow/core/kernels:gpu_util_hdrs",
81 "//tensorflow/core/kernels:ops_util_hdrs",
91 "kernels/fused_conv2d_bias_activation_op.cc",
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | BUILD | 2 # Contains experimental kernels for datasets and iterators.
32 "//tensorflow/core/kernels/data:captured_function",
33 "//tensorflow/core/kernels/data:dataset_utils",
34 "//tensorflow/core/kernels/data:take_dataset_op",
82 "//tensorflow/core/kernels/data:graph_rewrite_dataset",
95 "//tensorflow/core/kernels/data:captured_function",
108 "//tensorflow/core/kernels/data:captured_function",
109 "//tensorflow/core/kernels/data:window_dataset",
130 "//tensorflow/core/kernels/data:dataset_utils",
158 "//tensorflow/core/kernels:inplace_ops",
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | BUILD | 25 "//tensorflow/lite/kernels:kernel_util",
26 "//tensorflow/lite/kernels:op_macros",
27 "//tensorflow/lite/kernels:padding",
28 "//tensorflow/lite/kernels/internal:quantization_util",
29 "//tensorflow/lite/kernels/internal:reference_base",
30 "//tensorflow/lite/kernels/internal:tensor",
61 "//tensorflow/lite/kernels:kernel_util",
62 "//tensorflow/lite/kernels:op_macros",
63 "//tensorflow/lite/kernels:padding",
64 "//tensorflow/lite/kernels/internal:quantization_util",
[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/experimental/micro/tools/make/ |
| D | Makefile | 75 $(wildcard tensorflow/lite/experimental/micro/kernels/*test.cc)
82 $(wildcard tensorflow/lite/experimental/micro/kernels/*.cc) \
87 tensorflow/lite/kernels/kernel_util.cc \
88 tensorflow/lite/kernels/internal/quantization_util.cc
93 $(wildcard tensorflow/lite/experimental/micro/kernels/*.h) \
100 tensorflow/lite/kernels/kernel_util.h \
101 tensorflow/lite/kernels/op_macros.h \
102 tensorflow/lite/kernels/padding.h \
103 tensorflow/lite/kernels/internal/common.h \
104 tensorflow/lite/kernels/internal/compatibility.h \
[all …]
|
| /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/compiler/jit/ |
| D | BUILD | 54 "//tensorflow/compiler/jit/kernels:xla_ops",
55 "//tensorflow/compiler/tf2xla/kernels:xla_cpu_only_ops",
56 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
57 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
68 "//tensorflow/compiler/jit/kernels:xla_ops",
69 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
70 "//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
85 "//tensorflow/compiler/jit/kernels:xla_ops",
87 "//tensorflow/compiler/tf2xla/kernels:xla_ops",
104 "//tensorflow/compiler/jit/kernels:xla_ops",
[all …]
|
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| D | BUILD | 43 "kernels/mpi_ops.cc",
44 "kernels/ring.cc",
45 "kernels/ring.h",
49 "kernels/ring.cu.cc",
50 "kernels/ring.h",
62 "kernels/mpi_ops.cc",
63 "kernels/ring.cc",
66 "kernels/ring.h",
69 "kernels/ring.cu.cc",
101 kernels = [
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | BUILD | 21 "//tensorflow/lite/kernels/internal:types",
33 "//tensorflow/lite/kernels/internal:types",
54 "//tensorflow/lite/kernels:kernel_util",
55 "//tensorflow/lite/kernels:op_macros",
56 "//tensorflow/lite/kernels/internal:optimized",
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",
|
| /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.,
354 visibility = ["//tensorflow/core/kernels:friends"],
852 "//tensorflow/core/kernels:required",
1187 deps = ["//tensorflow/core/kernels:debug_ops"],
1253 "//tensorflow/core/kernels:bounds_check_lib",
[all …]
|
| /external/tensorflow/tensorflow/contrib/nearest_neighbor/ |
| D | BUILD | 24 "kernels/hyperplane_lsh_probes.cc",
45 kernels = [":nearest_neighbor_ops_kernels"],
57 srcs = ["kernels/hyperplane_lsh_probes.cc"],
69 hdrs = ["kernels/heap.h"],
75 srcs = ["kernels/heap_test.cc"],
79 "//tensorflow/core/kernels:ops_testutil",
85 hdrs = ["kernels/hyperplane_lsh_probes.h"],
95 srcs = ["kernels/hyperplane_lsh_probes_test.cc"],
101 "//tensorflow/core/kernels:ops_testutil",
|
| /external/tensorflow/tensorflow/contrib/resampler/ |
| D | BUILD | 22 kernels = [
44 "kernels/resampler_ops.cc",
45 "kernels/resampler_ops.h",
48 "kernels/resampler_ops_gpu.cu.cc",
49 "kernels/resampler_ops.h",
58 "//tensorflow/compiler/tf2xla/kernels:resampler_ops",
68 "kernels/resampler_ops.cc",
69 "kernels/resampler_ops.h",
73 "kernels/resampler_ops_gpu.cu.cc",
74 "kernels/resampler_ops.h",
[all …]
|