/**

* @license

* Copyright 2020 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 {Tensor} from '../../tensor';

import {convertToTensor} from '../../tensor_util_env';

import {TensorLike} from '../../types';

import {cast} from '../cast';

import {div} from '../div';

import {Reduction} from '../loss_ops_utils';

import {mean} from '../mean';

import {mul} from '../mul';

import {notEqual} from '../not_equal';

import {ones} from '../ones';

import {op} from '../operation';

import {scalar} from '../scalar';

import {sum} from '../sum';

/**

* Computes the weighted loss between two tensors.

*

* @param losses Tensor of shape `[batch_size, d1, ..., dN]`.

* @param weights Tensor whose rank is either 0, or the same rank as

* `losses`, and must be broadcastable to `losses` (i.e., all

* dimensions must be either `1`, or the same as the corresponding

* `losses` dimension).

*

* @doc {heading: 'Training', subheading: 'Losses', namespace: 'losses'}

*/

function computeWeightedLoss_<T extends Tensor, O extends Tensor>(

}

export const computeWeightedLoss = /* @__PURE__ */ op({computeWeightedLoss_});