gx优化

chenyc

2025-05-29 92f69c57b920cf62ecc9f15f9ed196fa26dbf2ac

1
2
3
4
5
6
7
8
9
/// <amd-module name="@tensorflow/tfjs-backend-webgl/dist/kernels/Cast" />
import { CastAttrs, CastInputs, KernelConfig, TensorInfo } from '@tensorflow/tfjs-core';
import { MathBackendWebGL } from '../backend_webgl';
export declare function cast(args: {
    inputs: CastInputs;
    backend: MathBackendWebGL;
    attrs: CastAttrs;
}): TensorInfo;
export declare const castConfig: KernelConfig;