Searched defs:TensorQueue (Results 1 – 1 of 1) sorted by relevance
130 struct TensorQueue : torch::CustomClassHolder { struct131 explicit TensorQueue(at::Tensor t) : init_tensor_(t) {} in TensorQueue() argument133 explicit TensorQueue(c10::Dict<std::string, at::Tensor> dict) { in TensorQueue() function148 c10::Dict<std::string, at::Tensor> serialize() const { in serialize()161 void push(at::Tensor x) { in push()168 at::Tensor pop() { in pop()180 at::Tensor top() { in top()189 int64_t size() { in size()193 bool is_empty() { in is_empty()198 double float_size() { in float_size()[all …]