Spaces:
Sleeping
Sleeping
| syntax = "proto3"; | |
| package tensorflow; | |
| option cc_enable_arenas = true; | |
| option java_outer_classname = "KernelDefProtos"; | |
| option java_multiple_files = true; | |
| option java_package = "org.tensorflow.framework"; | |
| 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; | |
| } | |