com.linkedin.feathr.common.tensorbuilder

Interface TensorBuilderFactory

All Known Implementing Classes:

DenseTensorBuilderFactory, UniversalTensorBuilderFactory

public interface TensorBuilderFactory

Method Summary

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

Method Detail

getTensorBuilder

TensorBuilder<?> getTensorBuilder(TensorType tensorType)

getBulkTensorBuilder

default BulkTensorBuilder getBulkTensorBuilder(TensorType tensorType)

getBulkTensorBuilder

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