pytorch  1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
  Fossies Dox: pytorch-1.8.2.tar.gz  ("unofficial" and yet experimental doxygen-generated source code documentation) File Reference
#include <memory>
#include <vector>
#include <gtest/gtest.h>
#include "caffe2/core/context_gpu.h"
#include "caffe2/operators/batch_matmul_op.h"
Include dependency graph for

Go to the source code of this file.


namespace  caffe2
 Copyright (c) 2016-present, Facebook, Inc.

Variable Documentation

◆ cuda_context_

std::unique_ptr<CUDAContext> cuda_context_

Definition at line 58 of file

◆ def_

◆ option_

DeviceOption option_

Definition at line 57 of file

Referenced by caffe2::Event::GetDeviceOption().

◆ ws_