blob: 57d8f4c46625352dc89565a63bf19eba9c0f9cd1 [file] [log] [blame]
op {
name: "SparseSoftmaxCrossEntropyWithLogits"
input_arg {
name: "features"
type_attr: "T"
}
input_arg {
name: "labels"
type_attr: "Tlabels"
}
output_arg {
name: "loss"
type_attr: "T"
}
output_arg {
name: "backprop"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tlabels"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
op {
name: "SparseSoftmaxCrossEntropyWithLogits"
input_arg {
name: "features"
type_attr: "T"
}
input_arg {
name: "labels"
type_attr: "Tlabels"
}
output_arg {
name: "loss"
type_attr: "T"
}
output_arg {
name: "backprop"
type_attr: "T"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_HALF
type: DT_BFLOAT16
type: DT_FLOAT
type: DT_DOUBLE
}
}
}
attr {
name: "Tlabels"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}