blob: 4fe13336ca4fecea69feff68f2f8782e67dc7d13 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
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;
}
|