Searched refs:kJitKernelLabel (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | disable_jit_test.cc | 67 Name("JitKernel").Device(DEVICE_CPU).Label(kJitKernelLabel),
74 {absl::StrCat("_kernel|string|'", kJitKernelLabel, "'")}, in TEST_F()
|
| D | op_kernel_test.cc | 892 Name("JitKernel").Device(DEVICE_CPU).Label(kJitKernelLabel),
898 {absl::StrCat("_kernel|string|'", kJitKernelLabel, "'")}, in TEST_F()
|
| D | op_kernel.cc | 64 const char* kJitKernelLabel = "JITCompiledKernel"; variable
1263 if (absl::StrContains(it->second.def.label(), kJitKernelLabel)) { in SetupOrDisableJit()
|
| D | op_kernel.h | 90 extern const char* kJitKernelLabel;
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_op.h | 197 .Label(kJitKernelLabel) additional_cstrs)
253 tf_op, platform, input_type, .Label(kJitKernelLabel) additional_cstrs)
310 tf_op, platform, input_type, .Label(kJitKernelLabel) additional_cstrs)
|