Spaces:
Build error
Build error
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; | |
} | |