Package org.nd4j.imports.graphmapper.tf.tensors

Class TFTensorMappers.Float16TensorMapper

java.lang.Object

org.nd4j.imports.graphmapper.tf.tensors.TFTensorMappers.BaseTensorMapper<float[],Buffer>

org.nd4j.imports.graphmapper.tf.tensors.TFTensorMappers.Float16TensorMapper

All Implemented Interfaces:

TFTensorMapper<float[],Buffer>

Enclosing class:

TFTensorMappers

public static class TFTensorMappers.Float16TensorMapper
extends TFTensorMappers.BaseTensorMapper<float[],Buffer>

Nested Class Summary

Nested classes/interfaces inherited from interface org.nd4j.imports.graphmapper.tf.tensors.TFTensorMapper

TFTensorMapper.ValueSource

Field Summary

Fields inherited from class org.nd4j.imports.graphmapper.tf.tensors.TFTensorMappers.BaseTensorMapper

tfTensor

Constructor Summary

Constructors

Constructor	Description
Float16TensorMapper(TensorProto tensorProto)

Method Summary

Modifier and Type	Method	Description
INDArray	arrayFor(long[] shape, float[] jArr)
Buffer	getBuffer(ByteBuffer bb)
void	getValue(float[] jArr, int i)
void	getValue(float[] jArr, Buffer buffer, int i)
float[]	newArray(int length)
int	valueCount()

Methods inherited from class org.nd4j.imports.graphmapper.tf.tensors.TFTensorMappers.BaseTensorMapper

dataType, isEmpty, shape, toNDArray, valueSource

Methods inherited from class java.lang.Object

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

Constructor Detail

Float16TensorMapper

public Float16TensorMapper(TensorProto tensorProto)

Method Detail

valueCount

public int valueCount()

newArray

public float[] newArray(int length)

getBuffer

public Buffer getBuffer(ByteBuffer bb)

getValue

public void getValue(float[] jArr,
                     int i)

getValue

public void getValue(float[] jArr,
                     Buffer buffer,
                     int i)

arrayFor

public INDArray arrayFor(long[] shape,
                         float[] jArr)