GetMinibatchesInCsrWithPhysicalReplica

public final class GetMinibatchesInCsrWithPhysicalReplica

Public Methods

static GetMinibatchesInCsrWithPhysicalReplica
create(Scope scope, Operand<String> programKey, Operand<Integer> rowIds, Operand<Integer> colIds, Operand<Float> gains, Operand<Long> splits, Operand<Integer> idCounts, Long sampleCount, Long numReplica, Long maxMinibatchesPerSc, Long maxIdsPerChipPerSample, Long tableVocabSize, Long featureWidth, Long numScPerChip, String tableName, String miniBatchInCsr)
Factory method to create a class wrapping a new GetMinibatchesInCsrWithPhysicalReplica operation.
Output<Integer>
idsUnpaddedSize()
Output<Integer>
numMinibatchesPerPhysicalSparseCore()
Output<Integer>
rowPointers()
Output<Integer>
rowPointersUnpaddedSize()
Output<Float>
sortedGains()
Output<Integer>
sortedSampleIds()
Output<Integer>
sortedTokenIds()

Inherited Methods

Public Methods

public static GetMinibatchesInCsrWithPhysicalReplica create (Scope scope, Operand<String> programKey, Operand<Integer> rowIds, Operand<Integer> colIds, Operand<Float> gains, Operand<Long> splits, Operand<Integer> idCounts, Long sampleCount, Long numReplica, Long maxMinibatchesPerSc, Long maxIdsPerChipPerSample, Long tableVocabSize, Long featureWidth, Long numScPerChip, String tableName, String miniBatchInCsr)

Factory method to create a class wrapping a new GetMinibatchesInCsrWithPhysicalReplica operation.

Parameters
scope current scope
Returns
  • a new instance of GetMinibatchesInCsrWithPhysicalReplica

public Output<Integer> idsUnpaddedSize ()

public Output<Integer> numMinibatchesPerPhysicalSparseCore ()

public Output<Integer> rowPointers ()

public Output<Integer> rowPointersUnpaddedSize ()

public Output<Float> sortedGains ()

public Output<Integer> sortedSampleIds ()

public Output<Integer> sortedTokenIds ()