Searched defs:TensorBuf (Results 1 – 3 of 3) sorted by relevance
29 class TensorBuf { class46 TensorBuf(Tensor* tensor, int dtype = DT_FLOAT, int size = 0) : tensor_(tensor) { in TensorBuf() function in TensorBuf
68 virtual void dumpBuffer(TensorBuf*, void*) {} in dumpBuffer() argument
74 virtual void dumpBuffer(TensorBuf*, void*) {} in dumpBuffer() argument