import { Order } from 'blockly/javascript';
/**
 * @category AI
 * @subcategory Operations
 * @module BasicOperations
 */
/**
 * Computes round of input tf.Tensor element-wise: round(x). It implements banker's rounding.
 *
 * @param {Tensor} tensor  The input tensor.
 * @returns {Tensor} tf.round(tensor)
 */
function roundOff(block, generator) {
    const tensor = generator.valueToCode(block, 'TENSOR', Order.NONE) || '';
    return [`tf.round(${tensor})`, Order.VOID];
}
export const basicBlockGenerator = {
    ['round_off_tensor']: roundOff,
};
//# sourceMappingURL=generators.js.map