Searched defs:kernel_def (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | execute.cc | 299 const KernelDef* kernel_def = nullptr; in GetKernelDef() local
308 const Device* op_device, const KernelDef* kernel_def, in IsHostMemoryArg()
884 const KernelDef* kernel_def) { in IntArgsAndRetvalsOnDevice()
958 EagerOperation* op, const KernelDef* kernel_def, in ExtractFunctionInputInfo()
1080 const KernelDef* kernel_def = nullptr; in GetOrCreateKernelAndDevice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 1321 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal()
1594 for (const auto& kernel_def : kernel_list.kernel()) { in LogAllRegisteredKernels() local
1610 const KernelDef& kernel_def = p.second.def; in GetFilteredRegisteredKernels() local
1627 for (const auto& kernel_def : kernel_list.kernel()) { in KernelsRegisteredForOp() local
1756 const KernelDef& kernel_def(key_registration.second.def); in ValidateKernelRegistrations() local
|
| H A D | kernel_def_util.cc | 36 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs, in KernelAttrsMatch()
|
| H A D | kernel_def_util_test.cc | 34 KernelDef kernel_def; in KernelDefFromText() local
|
| H A D | op_kernel_test_base.h | 153 const KernelDef* kernel_def = nullptr; variable
|
| H A D | op_kernel.h | 1510 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
1518 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | allowlisted_flex_ops_test.cc | 38 tensorflow::KernelDef kernel_def = kernel_list.kernel(i); in GetAllCpuKernels() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels.cc | 219 tensorflow::KernelDef* kernel_def = new tensorflow::KernelDef(); in TF_RegisterKernelBuilderWithKernelDef() local
|
| H A D | kernels_test.cc | 206 KernelDef kernel_def; in TEST() local
|