|
|
@@ -3197,7 +3197,8 @@
|
|
|
"type_param_str": "T",
|
|
|
"allowed_type_strs": [
|
|
|
"tensor(float)",
|
|
|
- "tensor(float16)"
|
|
|
+ "tensor(float16)",
|
|
|
+ "tensor(bfloat16)"
|
|
|
]
|
|
|
},
|
|
|
{
|
|
|
@@ -37450,7 +37451,8 @@
|
|
|
"type_param_str": "T",
|
|
|
"allowed_type_strs": [
|
|
|
"tensor(float)",
|
|
|
- "tensor(float16)"
|
|
|
+ "tensor(float16)",
|
|
|
+ "tensor(bfloat16)"
|
|
|
]
|
|
|
},
|
|
|
{
|
|
|
@@ -37458,7 +37460,8 @@
|
|
|
"type_param_str": "QK",
|
|
|
"allowed_type_strs": [
|
|
|
"tensor(float)",
|
|
|
- "tensor(float16)"
|
|
|
+ "tensor(float16)",
|
|
|
+ "tensor(bfloat16)"
|
|
|
]
|
|
|
},
|
|
|
{
|
|
|
@@ -53445,7 +53448,7 @@
|
|
|
"type": "string",
|
|
|
"required": false,
|
|
|
"default": "stretch",
|
|
|
- "description": "\nThis attribute describes how to interpret the `sizes` input with regard to keeping the original aspect ratio of the input, and it is not applicable when\nthe `scales` input is used. <br/>\n\nGiven a set of `sizes`, associated with a subset of `axes` (explicitly provided or default), and assuming `d = axes[i]`, with `i` being the index of the provided `sizes`. <br/>\n\nIf `keep_aspect_ratio_policy` is `\"stretch\"`, the original aspect ratio is disregarded, and the input is resized to the specified size: <br/>\n`out_size[d] = sizes[i]` <br/>\n\nIf `keep_aspect_ratio_policy` is `\"not_larger\"`, the sizes are adjusted so that no extent of the output is larger than the specified size, while keeping the original aspect ratio: <br/>\n`scale = Min(sizes[i] / in_size[d])` <br/>\n`out_size[d] = round_int(scale * in_size[i])` <br/>\n\nIf `keep_aspect_ratio_policy` is `\"not_smaller\"`, the sizes are adjusted so that no extent of the output is smaller than the specified size, while keeping the original aspect ratio: <br/>\n`scale = Max(sizes[i] / in_size[d])` <br/>\n`out_size[d] = round_int(scale * in_size[i])` <br/>\n\nFor non-resizable axes (those not specified in `axes`), the output size will be equal to the input size.\n\nNote: `round_int` stands for computing the nearest integer value, rounding halfway cases up."
|
|
|
+ "description": "\nThis attribute describes how to interpret the `sizes` input with regard to keeping the original aspect ratio of the input, and it is not applicable when\nthe `scales` input is used. <br/>\n\nGiven a set of `sizes`, associated with a subset of `axes` (explicitly provided or default), and assuming `d = axes[i]`, with `i` being the index of the provided `sizes`. <br/>\n\nIf `keep_aspect_ratio_policy` is `\"stretch\"`, the original aspect ratio is disregarded, and the input is resized to the specified size: <br/>\n`out_size[d] = sizes[i]` <br/>\n\nIf `keep_aspect_ratio_policy` is `\"not_larger\"`, the sizes are adjusted so that no extent of the output is larger than the specified size, while keeping the original aspect ratio: <br/>\n`scale = Min(sizes[i] / in_size[d])` <br/>\n`out_size[d] = round_int(scale * in_size[d])` <br/>\n\nIf `keep_aspect_ratio_policy` is `\"not_smaller\"`, the sizes are adjusted so that no extent of the output is smaller than the specified size, while keeping the original aspect ratio: <br/>\n`scale = Max(sizes[i] / in_size[d])` <br/>\n`out_size[d] = round_int(scale * in_size[d])` <br/>\n\nFor non-resizable axes (those not specified in `axes`), the output size will be equal to the input size.\n\nNote: `round_int` stands for computing the nearest integer value, rounding halfway cases up."
|
|
|
},
|
|
|
{
|
|
|
"name": "mode",
|
|
|
@@ -53582,7 +53585,7 @@
|
|
|
"type": "string",
|
|
|
"required": false,
|
|
|
"default": "stretch",
|
|
|
- "description": "\nThis attribute describes how to interpret the `sizes` input with regard to keeping the original aspect ratio of the input, and it is not applicable when\nthe `scales` input is used.\n\nGiven a set of `sizes`, associated with a subset of `axes` (explicitly provided or default), and assuming `d = axes[i]`, with `i` being the index of the provided `sizes`.\n\nIf `keep_aspect_ratio_policy` is `\"stretch\"`, the original aspect ratio is disregarded, and the input is resized to the specified size:\n`out_size[d] = sizes[i]`\n\nIf `keep_aspect_ratio_policy` is `\"not_larger\"`, the sizes are adjusted so that no extent of the output is larger than the specified size, while keeping the original aspect ratio:\n```\nscale = Min(sizes[i] / in_size[d])\nout_size[d] = round_int(scale * in_size[i])\n```\n\nIf `keep_aspect_ratio_policy` is `\"not_smaller\"`, the sizes are adjusted so that no extent of the output is smaller than the specified size, while keeping the original aspect ratio:\n```\nscale = Max(sizes[i] / in_size[d])\nout_size[d] = round_int(scale * in_size[i])\n```\n\nFor non-resizable axes (those not specified in `axes`), the output size will be equal to the input size.\n\nNote: `round_int` stands for computing the nearest integer value, rounding halfway cases up."
|
|
|
+ "description": "\nThis attribute describes how to interpret the `sizes` input with regard to keeping the original aspect ratio of the input, and it is not applicable when\nthe `scales` input is used.\n\nGiven a set of `sizes`, associated with a subset of `axes` (explicitly provided or default), and assuming `d = axes[i]`, with `i` being the index of the provided `sizes`.\n\nIf `keep_aspect_ratio_policy` is `\"stretch\"`, the original aspect ratio is disregarded, and the input is resized to the specified size:\n`out_size[d] = sizes[i]`\n\nIf `keep_aspect_ratio_policy` is `\"not_larger\"`, the sizes are adjusted so that no extent of the output is larger than the specified size, while keeping the original aspect ratio:\n```\nscale = Min(sizes[i] / in_size[d])\nout_size[d] = round_int(scale * in_size[d])\n```\n\nIf `keep_aspect_ratio_policy` is `\"not_smaller\"`, the sizes are adjusted so that no extent of the output is smaller than the specified size, while keeping the original aspect ratio:\n```\nscale = Max(sizes[i] / in_size[d])\nout_size[d] = round_int(scale * in_size[d])\n```\n\nFor non-resizable axes (those not specified in `axes`), the output size will be equal to the input size.\n\nNote: `round_int` stands for computing the nearest integer value, rounding halfway cases up."
|
|
|
},
|
|
|
{
|
|
|
"name": "mode",
|
|
|
@@ -59311,7 +59314,8 @@
|
|
|
"type_param_str": "T",
|
|
|
"allowed_type_strs": [
|
|
|
"tensor(float)",
|
|
|
- "tensor(float16)"
|
|
|
+ "tensor(float16)",
|
|
|
+ "tensor(bfloat16)"
|
|
|
]
|
|
|
},
|
|
|
{
|
Lutz Roeder