TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
cpid::SubBatchAsyncBatcher Member List

This is the complete list of members for cpid::SubBatchAsyncBatcher, including all inherited members.

accessMutex_cpid::AsyncBatcherprotected
allowPadding(bool allowPadding)cpid::SubBatchAsyncBatcherinline
allowPadding_cpid::SubBatchAsyncBatcherprotected
AsyncBatcher(ag::Container model, int batchsize, int padValue=-1, bool stripOutput=true, double stripValue=-1.)cpid::AsyncBatcher
batchedForward(ag::Variant state)cpid::AsyncBatchervirtual
batchReadyCV_cpid::AsyncBatcherprotected
batchReadyMutex_cpid::AsyncBatcherprotected
batchSize_cpid::AsyncBatcherprotected
consumeThread()cpid::AsyncBatcherprotected
consumeThread_cpid::AsyncBatcherprotected
consumeThreadStarted_cpid::AsyncBatcherprotected
findBatchInfo(ag::Variant const &batchInfoVar, std::string const &variableName)cpid::SubBatchAsyncBatcherstatic
forEachSubbatch(ag::Variant const &input, std::string const &inputName, torch::Tensor batchedInput, std::function< torch::Tensor(torch::Tensor)> do_fn=[](torch::Tensor t){return t;})cpid::SubBatchAsyncBatcherstatic
kBatchInfoKeycpid::SubBatchAsyncBatcherstatic
lockModel()cpid::AsyncBatcher
makeBatch(const std::vector< ag::Variant > &queries, double) overridecpid::SubBatchAsyncBatchervirtual
cpid::AsyncBatcher::makeBatch(const std::vector< ag::Variant > &queries)cpid::AsyncBatchervirtual
makeBatchTensors(std::vector< torch::Tensor > const &tensors, double padValue)cpid::SubBatchAsyncBatcher
model_cpid::AsyncBatcherprotected
modelMutex_cpid::AsyncBatcherprotected
padValue_cpid::AsyncBatcherprotected
queries_cpid::AsyncBatcherprotected
querySize_cpid::AsyncBatcherprotected
replies_cpid::AsyncBatcherprotected
setModel(ag::Container newModel)cpid::AsyncBatcher
shouldConsume()cpid::AsyncBatchervirtual
shouldStop_cpid::AsyncBatcherprotected
startBatching(int batchSize)cpid::AsyncBatcherprotected
stopBatching()cpid::AsyncBatcherprotected
stripOutput_cpid::AsyncBatcherprotected
stripValue_cpid::AsyncBatcherprotected
SubBatchAsyncBatcher(int batchSize, ag::Container model=nullptr)cpid::SubBatchAsyncBatcher
unBatch(const ag::Variant &out, bool stripOutput, double stripValue) overridecpid::SubBatchAsyncBatchervirtual
cpid::AsyncBatcher::unBatch(const ag::Variant &out)cpid::AsyncBatchervirtual
unBatchTensor(const torch::Tensor &out, std::vector< int64_t > const &batchSizes)cpid::SubBatchAsyncBatcherstatic
~AsyncBatcher()cpid::AsyncBatchervirtual
~SubBatchAsyncBatcher()cpid::SubBatchAsyncBatcher