Package com.linkedin.feathr.common.tensorbuilder

Interface TensorBuilderFactory

All Known Implementing Classes:

DenseTensorBuilderFactory, UniversalTensorBuilderFactory

public interface TensorBuilderFactory

Method Summary

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

Method Details

getTensorBuilder

TensorBuilder<?> getTensorBuilder(TensorType tensorType)

getBulkTensorBuilder

default BulkTensorBuilder getBulkTensorBuilder(TensorType tensorType)

getBulkTensorBuilder

default BulkTensorBuilder getBulkTensorBuilder(TensorType tensorType,
 long[] shape)