Searched defs:DeepAndWideFast (Results 1 – 1 of 1) sorted by relevance
38 struct DeepAndWideFast : torch::nn::Module { struct48 torch::Tensor forward( in forward()126 torch::Tensor mu_, sigma_, fc_w_, fc_b_, fc_w_t_;127 std::vector<torch::Tensor> prealloc_tensors;128 bool allocated = false;