Function int

tfjs-backend-webgpu/src/kernel_utils/int.ts:23–27

(input: TensorInfo, backend: WebGPUBackend)

import {UnaryOpProgram} from '../unary_op_webgpu';

export function int(input: TensorInfo, backend: WebGPUBackend): TensorInfo {
  const program = new UnaryOpProgram(input.shape, UnaryOpType.TO_INT);
  const output = backend.runWebGPUProgram(program, [input], 'int32');
  return {dataId: output.dataId, shape: output.shape, dtype: output.dtype};
}