提交 d9666eb3 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Update ops-related pbtxt files.

PiperOrigin-RevId: 306536433
Change-Id: I10ed89b9cc6983a8078e852afdbad6402d9fe6a7
上级 ce55348e
......@@ -308,3 +308,70 @@ op {
}
}
}
op {
name: "ArgMax"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BOOL
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
......@@ -308,3 +308,70 @@ op {
}
}
}
op {
name: "ArgMin"
input_arg {
name: "input"
type_attr: "T"
}
input_arg {
name: "dimension"
type_attr: "Tidx"
}
output_arg {
name: "output"
type_attr: "output_type"
}
attr {
name: "T"
type: "type"
allowed_values {
list {
type: DT_FLOAT
type: DT_DOUBLE
type: DT_INT32
type: DT_UINT8
type: DT_INT16
type: DT_INT8
type: DT_COMPLEX64
type: DT_INT64
type: DT_QINT8
type: DT_QUINT8
type: DT_QINT32
type: DT_BFLOAT16
type: DT_UINT16
type: DT_COMPLEX128
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BOOL
}
}
}
attr {
name: "Tidx"
type: "type"
default_value {
type: DT_INT32
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
attr {
name: "output_type"
type: "type"
default_value {
type: DT_INT64
}
allowed_values {
list {
type: DT_INT32
type: DT_INT64
}
}
}
}
......@@ -2153,6 +2153,7 @@ op {
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BOOL
}
}
}
......@@ -2219,6 +2220,7 @@ op {
type: DT_HALF
type: DT_UINT32
type: DT_UINT64
type: DT_BOOL
}
}
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册