Searched refs:kernel_def (Results 1 – 12 of 12) 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()
42 "KernelDef '", ProtoShortDebugString(kernel_def), in KernelAttrsMatch()
57 "KernelDef '", ProtoShortDebugString(kernel_def), in KernelAttrsMatch()
74 "OpKernel '", kernel_def.op(), "' has constraint on attr '", in KernelAttrsMatch()
76 "', KernelDef: '", ProtoShortDebugString(kernel_def), "'"); in KernelAttrsMatch()
|
| D | op_kernel.cc | 1068 void OpKernelRegistrar::InitInternal(const KernelDef* kernel_def, in InitInternal() argument
1072 if (kernel_def->op() != "_no_register") { in InitInternal()
1074 Key(kernel_def->op(), DeviceType(kernel_def->device_type()), in InitInternal()
1075 kernel_def->label()); in InitInternal()
1085 ->emplace(key, KernelRegistration(*kernel_def, kernel_class_name, in InitInternal()
1088 delete kernel_def; in InitInternal()
1209 for (const auto& kernel_def : kernel_list.kernel()) { in LogAllRegisteredKernels() local
1210 LOG(INFO) << "OpKernel ('" << ProtoShortDebugString(kernel_def) << "')"; in LogAllRegisteredKernels()
1224 const KernelDef& kernel_def = p.second.def; in GetFilteredRegisteredKernels() local
1225 if (predicate(kernel_def)) { in GetFilteredRegisteredKernels()
[all …]
|
| 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 | kernel_def_util.h | 26 Status KernelAttrsMatch(const KernelDef& kernel_def, AttrSlice attrs,
|
| D | op_kernel.h | 1454 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1458 if (kernel_def != nullptr) { in OpKernelRegistrar()
1459 InitInternal(kernel_def, kernel_class_name, std::move(factory)); in OpKernelRegistrar()
1465 OpKernelRegistrar(const KernelDef* kernel_def, StringPiece kernel_class_name, in OpKernelRegistrar() argument
1469 if (kernel_def != nullptr) { in OpKernelRegistrar()
1470 InitInternal(kernel_def, kernel_class_name, in OpKernelRegistrar()
1485 void InitInternal(const KernelDef* kernel_def, StringPiece kernel_class_name,
|
| D | op_kernel_test.cc | 504 const KernelDef* kernel_def = nullptr; local
507 FindKernelDef(device_type, def, &kernel_def, &kernel_class_name);
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_pb_text_files.txt | 11 tensorflow/core/framework/kernel_def.pb_text.cc
|
| D | tf_proto_files.txt | 15 tensorflow/core/framework/kernel_def.proto
|
| D | proto_text_pb_cc_files.txt | 11 tensorflow/core/framework/kernel_def.pb.cc
|
| D | proto_text_pb_h_files.txt | 11 tensorflow/core/framework/kernel_def.pb.h
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | tf_core_framework.cmake | 150 "tensorflow/core/framework/kernel_def.proto"
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 198 "framework/kernel_def.proto",
2110 proto_srcfile = "framework/kernel_def.proto",
|