/** * @license * Copyright 2021 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. * ============================================================================= */ /// import { ComplexInputs, KernelConfig, TensorInfo } from '@tensorflow/tfjs-core'; import { WebGPUBackend } from '../backend_webgpu'; /** * Complex tensors share data with their real and imaginary components. Complex * tensors' reference to the components is tracked by refCount on the individual * component. The refCounts are increased by the identity call. * * When a complex tensor is disposed, it will reduce the refCount on the * components by calling disposeData on each. */ export declare function complex(args: { inputs: ComplexInputs; backend: WebGPUBackend; }): TensorInfo; export declare const complexConfig: KernelConfig;