Package com.linkedin.feathr.common.tensorbuilder

Class DenseTensorBuilderFactory

java.lang.Object

com.linkedin.feathr.common.tensorbuilder.DenseTensorBuilderFactory

All Implemented Interfaces:

TensorBuilderFactory

public class DenseTensorBuilderFactory
extends Object
implements TensorBuilderFactory

Field Summary

Fields

Modifier and Type	Field	Description
static final DenseTensorBuilderFactory	INSTANCE

Constructor Summary

Constructors

Constructor	Description
DenseTensorBuilderFactory()

Method Summary

Modifier and Type	Method	Description
BulkTensorBuilder	getBulkTensorBuilder(TensorType tensorType)
DenseTensorBuilder	getBulkTensorBuilder(TensorType tensorType, long[] shape)
TensorBuilder<?>	getTensorBuilder(TensorType tensorType)

Methods inherited from class java.lang.Object

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

Field Details

INSTANCE

public static final DenseTensorBuilderFactory INSTANCE

Constructor Details

DenseTensorBuilderFactory

public DenseTensorBuilderFactory()

Method Details

getTensorBuilder

public TensorBuilder<?> getTensorBuilder(TensorType tensorType)

Specified by:

getTensorBuilder in interface TensorBuilderFactory

getBulkTensorBuilder

public BulkTensorBuilder getBulkTensorBuilder(TensorType tensorType)

Specified by:

getBulkTensorBuilder in interface TensorBuilderFactory

getBulkTensorBuilder

public DenseTensorBuilder getBulkTensorBuilder(TensorType tensorType,
 long[] shape)

Specified by:

getBulkTensorBuilder in interface TensorBuilderFactory