| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | allowlisted_flex_ops_test.cc | 48 std::set<std::string> all_kernels = GetAllCpuKernels(); in TEST() local
51 EXPECT_EQ(all_kernels.count(op_name), 1) in TEST()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | BUILD | 78 "//tensorflow/core:all_kernels",
102 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | BUILD | 154 "//tensorflow/core:all_kernels",
216 "//tensorflow/core:all_kernels",
319 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | BUILD | 55 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/transforms/ |
| D | BUILD | 185 "//tensorflow/core:all_kernels",
205 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/cc/ |
| D | BUILD | 90 "//tensorflow/core:all_kernels",
112 "//tensorflow/core:all_kernels",
147 "//tensorflow/core:all_kernels",
264 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | BUILD | 38 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 709 "//tensorflow/core:all_kernels",
2256 "//tensorflow/core:all_kernels",
2286 "//tensorflow/core:all_kernels",
2313 "//tensorflow/core:all_kernels",
2340 "//tensorflow/core:all_kernels",
2367 "//tensorflow/core:all_kernels",
2884 "//tensorflow/core:all_kernels",
2908 "//tensorflow/core:all_kernels",
2932 "//tensorflow/core:all_kernels",
2955 "//tensorflow/core:all_kernels",
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | BUILD | 65 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/runtime_fallback/ |
| D | BUILD | 54 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | build_defs.bzl | 322 all_kernels = aot_kernels + all_jit_kernels + all_paratial_jit_kernels
324 for (type, output_type, jit, jit_i64_indexed_for_large_tensors) in all_kernels:
413 for (type, output_type, jit, jit_i64_indexed_for_large_tensors) in all_kernels
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | BUILD | 145 "//tensorflow/core:all_kernels",
197 "//tensorflow/core:all_kernels",
331 "//tensorflow/core:all_kernels",
383 "//tensorflow/core:all_kernels",
1003 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 1259 std::unordered_multimap<string, KernelRegistration>& all_kernels = in SetupOrDisableJit() local
1261 auto it = all_kernels.begin(); in SetupOrDisableJit()
1262 while (it != all_kernels.end()) { in SetupOrDisableJit()
1278 it = all_kernels.erase(it); in SetupOrDisableJit()
1287 all_kernels.insert(std::move(jit_kernel)); in SetupOrDisableJit()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | BUILD | 53 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | BUILD | 60 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | BUILD | 76 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | BUILD | 238 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | BUILD | 197 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/python/client/ |
| D | BUILD | 219 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/data/ |
| D | BUILD | 430 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 1026 "//tensorflow/core:all_kernels",
1176 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 29 # ":all_kernels" - The cpu-specific kernels, plus gpu kernels if
32 # ":all_kernels", ":core", and a Session implementation.
750 name = "all_kernels",
773 ":all_kernels",
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | BUILD | 562 "//tensorflow/core:all_kernels",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 272 "//tensorflow/core:all_kernels",
642 "//tensorflow/core:all_kernels",
5300 "//tensorflow/core:all_kernels",
|