org.tensorflow.ndarray.impl.buffer.layout

Class Bfloat16Layout

java.lang.Object

org.tensorflow.ndarray.impl.buffer.layout.Bfloat16Layout

All Implemented Interfaces:

DataLayout<ShortDataBuffer,Float>, FloatDataLayout<ShortDataBuffer>

public final class Bfloat16Layout
extends Object
implements FloatDataLayout<ShortDataBuffer>

Data layout that converts 32-bit floats from/to 16-bit, truncating their mantissa to 7 bits but preserving the 8-bit exponent with the same bias.

Constructor Summary

Constructors

Constructor and Description
Bfloat16Layout()

Method Summary

Modifier and Type	Method and Description
float	readFloat(ShortDataBuffer buffer, long index) Reads n = scale() values from the buffer at the given index and return them as a float.
void	writeFloat(ShortDataBuffer buffer, float value, long index) Writes a float into the buffer at the given index after converting it to the buffer type.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.tensorflow.ndarray.buffer.layout.FloatDataLayout

applyTo, readObject, writeObject

Methods inherited from interface org.tensorflow.ndarray.buffer.layout.DataLayout

scale

Constructor Detail

Bfloat16Layout

public Bfloat16Layout()

Method Detail

writeFloat

public void writeFloat(ShortDataBuffer buffer,
                       float value,
                       long index)

Description copied from interface: FloatDataLayout

Writes a float into the buffer at the given index after converting it to the buffer type.

Specified by:

writeFloat in interface FloatDataLayout<ShortDataBuffer>

Parameters:

buffer - the buffer to write to

value - the float to convert and write

index - index in the buffer where the converted value should be written

See Also:

FloatDataLayout.writeObject(DataBuffer, Float, long)

readFloat

public float readFloat(ShortDataBuffer buffer,
                       long index)

Description copied from interface: FloatDataLayout

Reads n = scale() values from the buffer at the given index and return them as a float.

Specified by:

readFloat in interface FloatDataLayout<ShortDataBuffer>

Parameters:

buffer - the buffer to read from

index - position of the buffer to read in the buffer

Returns:

the float value

See Also:

FloatDataLayout.readObject(DataBuffer, long)