diff options
Diffstat (limited to 'tensorflow/core/ops/ops.pbtxt')
| -rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 299 |
1 files changed, 246 insertions, 53 deletions
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 36ecc0d9db..22a2f423c2 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -2490,8 +2490,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -2674,8 +2672,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -3989,8 +3985,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -4040,8 +4034,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -4140,6 +4132,8 @@ op { type: DT_INT32 type: DT_UINT8 type: DT_UINT16 + type: DT_UINT32 + type: DT_UINT64 type: DT_INT8 type: DT_INT16 type: DT_COMPLEX64 @@ -4165,6 +4159,8 @@ op { type: DT_INT32 type: DT_UINT8 type: DT_UINT16 + type: DT_UINT32 + type: DT_UINT64 type: DT_INT8 type: DT_INT16 type: DT_COMPLEX64 @@ -4340,6 +4336,34 @@ op { } } op { + name: "BoostedTreesCenterBias" + input_arg { + name: "tree_ensemble_handle" + type: DT_RESOURCE + } + input_arg { + name: "mean_gradients" + type: DT_FLOAT + } + input_arg { + name: "mean_hessians" + type: DT_FLOAT + } + input_arg { + name: "l1" + type: DT_FLOAT + } + input_arg { + name: "l2" + type: DT_FLOAT + } + output_arg { + name: "continue_centering" + type: DT_BOOL + } + is_stateful: true +} +op { name: "BoostedTreesCreateEnsemble" input_arg { name: "tree_ensemble_handle" @@ -4953,6 +4977,13 @@ op { name: "DstT" type: "type" } + attr { + name: "Truncate" + type: "bool" + default_value { + b: false + } + } } op { name: "Ceil" @@ -5702,8 +5733,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -5781,8 +5810,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -5860,8 +5887,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -8564,8 +8589,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -8636,8 +8659,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -8708,8 +8729,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -11457,7 +11476,9 @@ op { type: "type" allowed_values { list { + type: DT_HALF type: DT_FLOAT + type: DT_DOUBLE } } } @@ -11513,7 +11534,9 @@ op { type: "type" allowed_values { list { + type: DT_HALF type: DT_FLOAT + type: DT_DOUBLE } } } @@ -12342,29 +12365,6 @@ op { } } op { - name: "IdentityDataset" - input_arg { - name: "input_dataset" - type: DT_VARIANT - } - output_arg { - name: "handle" - type: DT_VARIANT - } - attr { - name: "output_types" - type: "list(type)" - has_minimum: true - minimum: 1 - } - attr { - name: "output_shapes" - type: "list(shape)" - has_minimum: true - minimum: 1 - } -} -op { name: "IdentityN" input_arg { name: "input" @@ -12457,7 +12457,6 @@ op { name: "Tout" type: "list(type)" has_minimum: true - minimum: 1 } attr { name: "then_branch" @@ -13237,6 +13236,36 @@ op { is_stateful: true } op { + name: "IteratorFromStringHandleV2" + input_arg { + name: "string_handle" + type: DT_STRING + } + output_arg { + name: "resource_handle" + type: DT_RESOURCE + } + attr { + name: "output_types" + type: "list(type)" + default_value { + list { + } + } + has_minimum: true + } + attr { + name: "output_shapes" + type: "list(shape)" + default_value { + list { + } + } + has_minimum: true + } + is_stateful: true +} +op { name: "IteratorGetNext" input_arg { name: "iterator" @@ -13297,6 +13326,34 @@ op { is_stateful: true } op { + name: "IteratorV2" + output_arg { + name: "handle" + type: DT_RESOURCE + } + attr { + name: "shared_name" + type: "string" + } + attr { + name: "container" + type: "string" + } + attr { + name: "output_types" + type: "list(type)" + has_minimum: true + minimum: 1 + } + attr { + name: "output_shapes" + type: "list(shape)" + has_minimum: true + minimum: 1 + } + is_stateful: true +} +op { name: "L2Loss" input_arg { name: "t" @@ -15443,8 +15500,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -15522,8 +15577,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -15594,8 +15647,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -15737,8 +15788,6 @@ op { list { s: "NHWC" s: "NCHW" - s: "HWNC" - s: "HWCN" } } } @@ -16958,6 +17007,33 @@ op { } } op { + name: "NonMaxSuppressionWithOverlaps" + input_arg { + name: "overlaps" + type: DT_FLOAT + } + input_arg { + name: "scores" + type: DT_FLOAT + } + input_arg { + name: "max_output_size" + type: DT_INT32 + } + input_arg { + name: "overlap_threshold" + type: DT_FLOAT + } + input_arg { + name: "score_threshold" + type: DT_FLOAT + } + output_arg { + name: "selected_indices" + type: DT_INT32 + } +} +op { name: "NotEqual" input_arg { name: "x" @@ -26113,6 +26189,7 @@ op { type: DT_HALF type: DT_UINT32 type: DT_UINT64 + type: DT_BOOL } } } @@ -27339,6 +27416,17 @@ op { } } op { + name: "SinkDataset" + input_arg { + name: "input_dataset" + type: DT_VARIANT + } + output_arg { + name: "handle" + type: DT_VARIANT + } +} +op { name: "Size" input_arg { name: "input" @@ -27502,7 +27590,11 @@ op { type: DT_INT64 } input_arg { - name: "stride" + name: "window_shift" + type: DT_INT64 + } + input_arg { + name: "window_stride" type: DT_INT64 } output_arg { @@ -30138,6 +30230,54 @@ op { } } op { + name: "SparseSliceGrad" + input_arg { + name: "backprop_val_grad" + type_attr: "T" + } + input_arg { + name: "input_indices" + type: DT_INT64 + } + input_arg { + name: "input_start" + type: DT_INT64 + } + input_arg { + name: "output_indices" + type: DT_INT64 + } + output_arg { + name: "val_grad" + type_attr: "T" + } + 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 + } + } + } +} +op { name: "SparseSoftmax" input_arg { name: "sp_indices" @@ -31170,6 +31310,32 @@ op { is_stateful: true } op { + name: "StatefulPartitionedCall" + input_arg { + name: "args" + type_list_attr: "Tin" + } + output_arg { + name: "output" + type_list_attr: "Tout" + } + attr { + name: "Tin" + type: "list(type)" + has_minimum: true + } + attr { + name: "Tout" + type: "list(type)" + has_minimum: true + } + attr { + name: "f" + type: "func" + } + is_stateful: true +} +op { name: "StatelessMultinomial" input_arg { name: "logits" @@ -35070,6 +35236,33 @@ op { is_stateful: true } op { + name: "WindowDataset" + input_arg { + name: "input_dataset" + type: DT_VARIANT + } + input_arg { + name: "window_size" + type: DT_INT64 + } + output_arg { + name: "handle" + type: DT_VARIANT + } + attr { + name: "output_types" + type: "list(type)" + has_minimum: true + minimum: 1 + } + attr { + name: "output_shapes" + type: "list(shape)" + has_minimum: true + minimum: 1 + } +} +op { name: "WriteAudioSummary" input_arg { name: "writer" |