aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/framework/kernel_def.proto
diff options
context:
space:
mode:
authorGravatar Manjunath Kudlur <keveman@gmail.com>2015-11-06 16:27:58 -0800
committerGravatar Manjunath Kudlur <keveman@gmail.com>2015-11-06 16:27:58 -0800
commitf41959ccb2d9d4c722fe8fc3351401d53bcf4900 (patch)
treeef0ca22cb2a5ac4bdec9d080d8e0788a53ed496d /tensorflow/core/framework/kernel_def.proto
TensorFlow: Initial commit of TensorFlow library.
TensorFlow is an open source software library for numerical computation using data flow graphs. Base CL: 107276108
Diffstat (limited to 'tensorflow/core/framework/kernel_def.proto')
-rw-r--r--tensorflow/core/framework/kernel_def.proto33
1 files changed, 33 insertions, 0 deletions
diff --git a/tensorflow/core/framework/kernel_def.proto b/tensorflow/core/framework/kernel_def.proto
new file mode 100644
index 0000000000..db7856a156
--- /dev/null
+++ b/tensorflow/core/framework/kernel_def.proto
@@ -0,0 +1,33 @@
+syntax = "proto3";
+
+package tensorflow;
+// option cc_enable_arenas = true;
+
+import "tensorflow/core/framework/attr_value.proto";
+
+message KernelDef {
+ // Must match the name of an Op.
+ string op = 1;
+
+ // Type of device this kernel runs on.
+ string device_type = 2;
+
+ message AttrConstraint {
+ // Name of an attr from the Op.
+ string name = 1;
+
+ // A list of values that this kernel supports for this attr.
+ // Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
+ AttrValue allowed_values = 2;
+ }
+ repeated AttrConstraint constraint = 3;
+
+ // Names of the Op's input_/output_args that reside in host memory
+ // instead of device memory.
+ repeated string host_memory_arg = 4;
+
+ // This allows experimental kernels to be registered for an op that
+ // won't be used unless the user specifies a "_kernel" attr with
+ // value matching this.
+ string label = 5;
+}