/**
|
* @license
|
* Copyright 2018 Google Inc. 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.
|
* =============================================================================
|
*/
|
import { GPGPUProgram } from './gpgpu_math';
|
export declare const DIV = "\n // vec4 one = vec4(equal(a, b));\n // return one + (vec4(1.0) - one) * a / b;\n vec4 result = a / b;\n if(a.x == b.x) {\n result.x = 1.;\n }\n if(a.y == b.y) {\n result.y = 1.;\n }\n if(a.z == b.z) {\n result.z = 1.;\n }\n if(a.w == b.w) {\n result.w = 1.;\n }\n\n return result;\n";
|
export declare const INT_DIV = "\n ivec4 ia = round(a);\n ivec4 ib = round(b);\n bvec4 cond = notEqual(ib, ivec4(0));\n ivec4 result = ivec4(0);\n vec4 s = sign(a) * sign(b);\n\n // Windows (D3D) wants guaranteed non-zero int division at compile-time.\n if (cond[0]) {\n result[0] = idiv(ia[0], ib[0], s[0]);\n }\n if (cond[1]) {\n result[1] = idiv(ia[1], ib[1], s[1]);\n }\n if (cond[2]) {\n result[2] = idiv(ia[2], ib[2], s[2]);\n }\n if (cond[3]) {\n result[3] = idiv(ia[3], ib[3], s[3]);\n }\n return vec4(result);\n";
|
export declare const POW: string;
|
export declare const PRELU = "\n vec4 aLessThanZero = vec4(lessThan(a, vec4(0.)));\n return (aLessThanZero * (b * a)) + ((vec4(1.0) - aLessThanZero) * a);\n";
|
export declare const ELU_DER = "\n vec4 bGTEZero = vec4(greaterThanEqual(b, vec4(0.)));\n return (bGTEZero * a) + ((vec4(1.0) - bGTEZero) * (a * (b + vec4(1.0))));\n";
|
export declare const ATAN2: string;
|
export declare const EQUAL = "\n return vec4(equal(a, b));\n";
|
export declare const NOT_EQUAL = "\n return vec4(notEqual(a, b));\n";
|
export declare const LESS = "\n return vec4(lessThan(a, b));\n";
|
export declare const LESS_EQUAL = "\n return vec4(lessThanEqual(a, b));\n";
|
export declare const GREATER = "\n return vec4(greaterThan(a, b));\n";
|
export declare const GREATER_EQUAL = "\n return vec4(greaterThanEqual(a, b));\n";
|
export declare const LOGICAL_AND = "\n return vec4(\n vec4(greaterThanEqual(a, vec4(1.0))) *\n vec4(greaterThanEqual(b, vec4(1.0))));\n";
|
export declare const LOGICAL_OR = "\n return min(\n vec4(greaterThanEqual(a, vec4(1.0))) +\n vec4(greaterThanEqual(b, vec4(1.0))),\n vec4(1.0));\n";
|
export declare const MAX: string;
|
export declare const MIN: string;
|
export declare const MOD: string;
|
export declare class BinaryOpPackedProgram implements GPGPUProgram {
|
variableNames: string[];
|
outputShape: number[];
|
userCode: string;
|
supportsBroadcasting: boolean;
|
packedInputs: boolean;
|
packedOutput: boolean;
|
constructor(op: string, aShape: number[], bShape: number[], checkOutOfBounds?: boolean);
|
}
|
