Searched refs:kernel_def (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | kernel_def_util.cc | 36 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs, in KernelAttrsMatch() argument
39 for (const auto& constraint : kernel_def.constraint()) { in KernelAttrsMatch()
61 "KernelDef '", kernel_def.ShortDebugString(), in KernelAttrsMatch()
68 "KernelDef '", kernel_def.ShortDebugString(), in KernelAttrsMatch()
77 "OpKernel '", kernel_def.op(), "' has constraint on attr '", in KernelAttrsMatch()
79 "', KernelDef: '", kernel_def.ShortDebugString(), "'"); in KernelAttrsMatch()
88 "KernelDef '", kernel_def.ShortDebugString(), \ in KernelAttrsMatch()
125 "KernelDef '", kernel_def.ShortDebugString(), in KernelAttrsMatch()
|
| D | kernel_def_util_test.cc | 34 KernelDef kernel_def; in KernelDefFromText() local
35 EXPECT_TRUE(protobuf::TextFormat::MergeFromString(text, &kernel_def)); in KernelDefFromText()
36 return kernel_def; in KernelDefFromText()
|
| D | op_kernel.cc | 1256 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal() argument
1260 Key(kernel_def->op(), DeviceType(kernel_def->device_type()), in InitInternal()
1261 kernel_def->label()); in InitInternal()
1275 KernelRegistration(*kernel_def, kernel_class_name, std::move(factory))); in InitInternal()
1276 delete kernel_def; in InitInternal()
1526 for (const auto& kernel_def : kernel_list.kernel()) { in LogAllRegisteredKernels() local
1527 LOG(INFO) << "OpKernel ('" << kernel_def.ShortDebugString() << "')"; in LogAllRegisteredKernels()
1542 const KernelDef& kernel_def = p.second.def; in GetFilteredRegisteredKernels() local
1543 if (predicate(kernel_def)) { in GetFilteredRegisteredKernels()
1544 *kernel_list.add_kernel() = kernel_def; in GetFilteredRegisteredKernels()
[all …]
|
| D | kernel_def_util.h | 26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
|
| D | op_kernel.h | 1456 << ([](::tensorflow::KernelDef const* kernel_def) { \
1458 kernel_def, #__VA_ARGS__, \
1541 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1543 InitInternal(kernel_def, kernel_class_name, std::move(factory)); in OpKernelRegistrar()
1548 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1550 InitInternal(kernel_def, kernel_class_name, in OpKernelRegistrar()
1564 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
| D | BUILD | 127 "kernel_def.proto",
1338 proto_srcfile = "kernel_def.proto",
1502 srcs = ["kernel_def.proto"],
|
| D | op_kernel_test.cc | 579 const KernelDef* kernel_def = nullptr; local
582 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | allowlisted_flex_ops_test.cc | 37 tensorflow::KernelDef kernel_def = kernel_list.kernel(i); in GetAllCpuKernels() local
38 result.insert(kernel_def.op()); in GetAllCpuKernels()
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 201 "//tensorflow/core/framework:kernel_def.proto",
|