batch_matmul_op_gpu_test.cc 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 batch_matmul_op_gpu_test.cc:

Go to the source code of this file.

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

Variable Documentation

cuda_context_

std::unique_ptr<CUDAContext> cuda_context_

protected

Definition at line 58 of file batch_matmul_op_gpu_test.cc.

def_

OperatorDef def_

protected

Definition at line 60 of file batch_matmul_op_gpu_test.cc.

Referenced by caffe2::GradientMakerBase::Def(), caffe2::TensorInferenceFunction::GetFCGradient::GetGradientDefs(), caffe2::GradientMakerBase::GI(), caffe2::GradientMakerBase::GI_I(), caffe2::GradientMakerBase::GI_V(), caffe2::GradientMakerBase::GO(), caffe2::GradientMakerBase::GO_I(), caffe2::GradientMakerBase::GO_V(), caffe2::GradientMakerBase::I(), caffe2::NumOutputs(), caffe2::GradientMakerBase::O(), caffe2::OPERATOR_SCHEMA(), caffe2::GradientMakerBase::SetDense(), caffe2::GradientMakerBase::SetSparse(), and caffe2::GradientMakerBase::VerifyOp().

option_

DeviceOption option_

protected

Definition at line 57 of file batch_matmul_op_gpu_test.cc.

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

ws_

Workspace ws_

protected

Definition at line 59 of file batch_matmul_op_gpu_test.cc.

Referenced by caffe2::ConvDNNLowPOp< std::uint8_t, false >::PartitionGroupedNHWCConv_(), caffe2::SaveOp< Context >::RunOnDevice(), caffe2::ConvOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvTransposeOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::DeformConvOp< T, Context >::RunOnDeviceWithOrderNCHW(), caffe2::ConvOp< T, Context >::RunOnDeviceWithOrderNHWC(), and caffe2::ConvTransposeOp< T, Context >::RunOnDeviceWithOrderNHWC().