/**
|
* @license
|
* Copyright 2018 Google LLC. All Rights Reserved.
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
*
|
* http://www.apache.org/licenses/LICENSE-2.0
|
*
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
* =============================================================================
|
*/
|
/// <amd-module name="@tensorflow/tfjs-backend-webgl/dist/shader_compiler_util" />
|
/**
|
* Produces GLSL code that derives logical coordinates from a flat
|
* index. The code performs integer division with each stride and decrements
|
* the index until the index equals the final dimension coordinate.
|
*/
|
export declare function getLogicalCoordinatesFromFlatIndex(coords: string[], shape: number[], index?: string): string;
|
export declare function getOutputLogicalCoordinatesFromFlatIndexByUniform(coords: string[], shape: number[], index?: string): string;
|
export declare function getLogicalCoordinatesFromFlatIndexByUniform(coords: string[], variableName: string, index?: string): string;
|
/**
|
* Produces GLSL code that computes the dot product of the input x and y
|
* vectors. Handles splitting inputs into increments of vec4s when necessary.
|
*/
|
export declare function dotify(x: string[], y: string[]): string;
|
/**
|
* Produces GLSL that computes the flat index from 3D coordinates.
|
*/
|
export declare function getFlatIndexFrom3D(shape: [number, number, number]): string;
|
export declare function getFlatIndexFrom3DOutput(): string;
|
export declare const ENCODE_FLOAT_SNIPPET = "\n const float FLOAT_MAX = 1.70141184e38;\n const float FLOAT_MIN = 1.17549435e-38;\n\n lowp vec4 encode_float(highp float v) {\n if (isnan(v)) {\n return vec4(255, 255, 255, 255);\n }\n\n highp float av = abs(v);\n\n if(av < FLOAT_MIN) {\n return vec4(0.0, 0.0, 0.0, 0.0);\n } else if(v > FLOAT_MAX) {\n return vec4(0.0, 0.0, 128.0, 127.0) / 255.0;\n } else if(v < -FLOAT_MAX) {\n return vec4(0.0, 0.0, 128.0, 255.0) / 255.0;\n }\n\n highp vec4 c = vec4(0,0,0,0);\n\n highp float e = floor(log2(av));\n highp float m = exp2(fract(log2(av))) - 1.0;\n\n c[2] = floor(128.0 * m);\n m -= c[2] / 128.0;\n c[1] = floor(32768.0 * m);\n m -= c[1] / 32768.0;\n c[0] = floor(8388608.0 * m);\n\n highp float ebias = e + 127.0;\n c[3] = floor(ebias / 2.0);\n ebias -= c[3] * 2.0;\n c[2] += floor(ebias) * 128.0;\n\n c[3] += 128.0 * step(0.0, -v);\n\n return c / 255.0;\n }\n";
|
