saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-15 21:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2/operators
History
Jerry Zhang 9f4bcdf075 caffe2::DeviceType -> at::DeviceType (#11254) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11254
Previously we use DeviceType in caffe2.proto directly, but it's an `enum` and have implicit conversion to int, which does not have type safety, e.g. we have to explicitly check for a device type is valid in event.h:
```
template <int d>
struct EventCreateFunctionRegisterer {
  explicit EventCreateFunctionRegisterer(EventCreateFunction f) {
    static_assert(d < MaxDeviceTypes, "");
    Event::event_creator_[d] = f;
  }
};
```
at::DeviceType is an `enum class`, and it does not have implicit conversion to int, and provides better type safety guarantees. In this diff we have done the following refactor(taking CPU as an example):

    1. caffe2::DeviceType → caffe2::DeviceTypeProto
    2. caffe2::CPU → caffe2::PROTO_CPU
    3. caffe2::DeviceType = at::DeviceType
    4. caffe2::CPU = at::DeviceType::CPU

codemod -d caffe2/caffe2 --extensions h,cc,cpp 'device_type\(\), ' 'device_type(), PROTO_'
+ some manual changes

In short, after this diff, in c++, caffe2::CPU refers to the at::DeviceType::CPU and the old proto caffe2::CPU will be caffe2::PROTO_CPU.
In python side, we have a temporary workaround that alias `caffe2_pb2.CPU = caffe2_pb2.PROOT_CPU` to make the change easier to review and this will be removed later.

Reviewed By: ezyang

Differential Revision: D9545704

fbshipit-source-id: 461a28a4ca74e616d3ee183a607078a717fd38a7
2018-09-05 16:28:09 -07:00
..
experimental/c10 Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
hip Update hip files (#9826) 2018-07-27 16:54:39 -07:00
rnn caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
abs_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
abs_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
abs_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
accumulate_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
accumulate_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
accumulate_op.h Optimize WeightedSumOp for two inputs (#11049) 2018-09-01 11:54:55 -07:00
accuracy_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
accuracy_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
accuracy_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
acos_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
acos_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
acos_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
activation_ops_cudnn.h Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
affine_channel_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
affine_channel_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
affine_channel_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
apmeter_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
apmeter_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
arg_ops.cc Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
arg_ops.cu Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
arg_ops.h Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
asin_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
asin_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
asin_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
assert_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
assert_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
assert_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
atan_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
atan_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
atan_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
atomic_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
batch_box_cox_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_box_cox_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
batch_bucketize_op.cc Add BatchBucketizeOp in caffe2 (#9385) 2018-07-13 20:39:30 -07:00
batch_bucketize_op.h Add BatchBucketizeOp in caffe2 (#9385) 2018-07-13 20:39:30 -07:00
batch_gather_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_gather_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
batch_gather_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
batch_matmul_op.cc Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
batch_matmul_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_matmul_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
batch_matmul_op_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
batch_matmul_op_test.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
batch_moments_op.cc Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
batch_moments_op.cu Add implementation for batch_moments_op (#9510) 2018-07-18 11:59:54 -07:00
batch_moments_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
batch_sparse_to_dense_op.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
batch_sparse_to_dense_op.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
bbox_transform_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
bbox_transform_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
bisect_percentile_op.cc Add a bisect percentile operator (#10563) 2018-08-20 13:14:05 -07:00
bisect_percentile_op.h Add a bisect percentile operator (#10563) 2018-08-20 13:14:05 -07:00
boolean_mask_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
boolean_mask_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
boolean_mask_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
boolean_unmask_ops.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
boolean_unmask_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
boolean_unmask_ops.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
boolean_unmask_ops_test.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
box_with_nms_limit_op.cc Fixed a bug in box_with_nms_limit where it may produce more bounding boxes than specified. (#10390) 2018-08-14 23:54:23 -07:00
box_with_nms_limit_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
byte_weight_dequant_op.cc Add byte_weight_dequant_op 2018-07-18 16:27:21 -07:00
byte_weight_dequant_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
cast_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
cast_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cast_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
cbrt_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
cbrt_op.cu Add Cube and Cbrt Ops (#8991) 2018-06-28 14:55:30 -07:00
cbrt_op.h Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
ceil_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
ceil_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
ceil_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
channel_backprop_stats_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_backprop_stats_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_backprop_stats_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_shuffle_op.cc Update CopyMatrix in math (#9792) 2018-07-25 16:10:52 -07:00
channel_shuffle_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
channel_shuffle_op_gpu.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_stats_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_stats_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
channel_stats_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
clip_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
clip_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
clip_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
CMakeLists.txt Implement c10 ops needed for benchmark (#9360) 2018-08-03 16:09:27 -07:00
collect_and_distribute_fpn_rpn_proposals_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
collect_and_distribute_fpn_rpn_proposals_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
communicator_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
communicator_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
concat_split_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
concat_split_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
concat_split_op_gpu.cc Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
conditional_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conditional_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_gradient_op.cc Add cost inference to ConvGradient and WeightedSum operators (#10744) 2018-09-05 13:56:05 -07:00
conv_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
conv_op.h group conv with NHWC layout (#10585) 2018-08-17 00:39:23 -07:00
conv_op_cache_cudnn.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_op_cache_cudnn.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_op_cache_cudnn_test.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_op_cudnn.cc fix bug in 3d group convolution (#9860) 2018-08-01 22:55:38 -07:00
conv_op_eigen.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
conv_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_op_impl.h group conv with NHWC layout (#10585) 2018-08-17 00:39:23 -07:00
conv_op_shared.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_op_shared.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_op_shared_gpu.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_pool_op_base.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
conv_transpose_gradient_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_transpose_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
conv_transpose_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
conv_transpose_op_cudnn.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_transpose_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_transpose_op_impl.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_transpose_op_mobile.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
conv_transpose_op_mobile.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_transpose_op_mobile_impl.h Update Im2Col related to make preparation for group conv in NHWC order. (#10439) 2018-08-15 17:10:24 -07:00
conv_transpose_op_mobile_test.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
conv_transpose_unpool_op_base.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
cos_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
cos_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
cos_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
cosh_op.cc Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
cosh_op.cu Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
cosh_op.h Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
cosine_embedding_criterion_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
cosine_embedding_criterion_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
cosine_embedding_criterion_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
counter_ops.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
counter_ops.h More changes for hidden visibility (#10692) 2018-08-21 13:39:57 -07:00
counter_ops_gpu.cc HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
create_scope_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
create_scope_op.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
cross_entropy_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
cross_entropy_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
cross_entropy_op.h Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
ctc_beam_search_decoder_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
ctc_beam_search_decoder_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
ctc_greedy_decoder_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
ctc_greedy_decoder_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
cube_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
cube_op.cu Add Cube and Cbrt Ops (#8991) 2018-06-28 14:55:30 -07:00
cube_op.h Add Cube and Cbrt Ops (#8991) 2018-06-28 14:55:30 -07:00
dataset_ops.cc Cleaner semantics for Reserve (#10261) 2018-08-06 14:40:16 -07:00
dataset_ops.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
deform_conv_gradient_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
deform_conv_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
deform_conv_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
deform_conv_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
deform_conv_op_impl.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
depthwise_3x3_conv_op_cudnn.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
distance_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
distance_op.cu Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
distance_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
do_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
do_op.h Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
do_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
dropout_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
dropout_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
dropout_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
dropout_op_cudnn.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
elementwise_add_gradient_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_add_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_add_op.h Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
elementwise_add_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
elementwise_div_gradient_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
elementwise_div_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_div_op.cu Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
elementwise_div_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
elementwise_linear_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
elementwise_linear_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
elementwise_linear_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
elementwise_logical_ops.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
elementwise_logical_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
elementwise_mul_gradient_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_mul_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_mul_op.cu Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
elementwise_mul_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
elementwise_op_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
elementwise_op_test.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
elementwise_op_test.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
elementwise_ops.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
elementwise_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
elementwise_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
elementwise_ops_schema.cc Add in_place test for binary ops 2018-06-28 11:45:35 -07:00
elementwise_ops_utils.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
elementwise_ops_utils.h Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
elementwise_sub_gradient_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_sub_op.cc [caffe2] Move elementwise grad ops to separate files (#8315) 2018-06-11 15:38:36 -07:00
elementwise_sub_op.h Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
elementwise_sub_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
elementwise_sum_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
elu_op.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
elu_op.cu Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
elu_op.h Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
elu_op_cudnn.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
enforce_finite_op.cc Update from Facebook (#6692) 2018-04-17 23:36:40 -07:00
enforce_finite_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
enforce_finite_op.h Fix EnforceFiniteOp 2018-07-27 22:01:23 -07:00
ensure_clipped_op.cc Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
ensure_clipped_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
ensure_cpu_output_op.cc Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
ensure_cpu_output_op.cu Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
ensure_cpu_output_op.h Fix EnsureCPUOutputOp (#10651) 2018-08-20 12:12:48 -07:00
exp_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
exp_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
exp_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
expand_op.cc In Expand support using -1 to indicate preserving original size (#10174) 2018-08-02 22:09:47 -07:00
expand_op.h Update broadcast with alpha to reduce num of launching kernels. (#10235) 2018-08-04 19:54:20 -07:00
expand_op_gpu.cc New operator 'expand' (#8263) 2018-06-18 16:33:47 -07:00
expand_squeeze_dims_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
expand_squeeze_dims_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
expand_squeeze_dims_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
extend_tensor_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
fc_inference.cc Update from facebook (#7696) 2018-05-19 23:10:48 -07:00
fc_inference.h Integrate fbgemm fp16 with Caffe2 2018-03-27 18:10:39 -07:00
feature_maps_ops.cc [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
feature_maps_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
feed_blob_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
feed_blob_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
filler_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
filler_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
filler_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
find_duplicate_elements_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
find_duplicate_elements_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
find_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
find_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
find_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
flatten_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
flatten_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
flexible_top_k.cc Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
flexible_top_k.h
floor_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
floor_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
floor_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
free_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
free_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
free_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
fully_connected_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
fully_connected_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
fully_connected_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
fused_rowwise_8bit_conversion_ops.cc Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
fused_rowwise_8bit_conversion_ops.h Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
fused_rowwise_random_quantization_ops.cc Faster random number generation in fused_rowwise_random_quantization_ops (#10634) 2018-08-22 13:15:46 -07:00
fused_rowwise_random_quantization_ops.h Faster random number generation in fused_rowwise_random_quantization_ops (#10634) 2018-08-22 13:15:46 -07:00
gather_fused_8bit_rowwise_op.cc add GatherFused8BitRowwise operator (#2167) 2018-03-09 07:42:17 -08:00
gather_fused_8bit_rowwise_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
gather_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
gather_op.cu IsType<TensorCPU> -> IsType<Tensor>(CPU) (#10135) 2018-08-03 17:24:59 -07:00
gather_op.h Fix bugs in handling of negative slice + gather indices (#10973) 2018-08-28 23:40:40 -07:00
gather_ranges_to_dense_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
gather_ranges_to_dense_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
generate_proposals_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
generate_proposals_op.h More fixes for hidden visibility (#10624) 2018-08-20 10:11:59 -07:00
generate_proposals_op_test.cc Avoid shadowing i, j vars in GeneralProposals test (#10721) 2018-08-21 09:11:38 -07:00
generate_proposals_op_util_boxes.h Fix some warnings (#10297) 2018-08-14 10:40:08 -07:00
generate_proposals_op_util_boxes_test.cc Clip horizontal bounding boxes during rotated detection for backward compatibility (#9403) 2018-07-16 20:24:49 -07:00
generate_proposals_op_util_nms.h fix the reference link path 2018-07-31 09:09:46 -07:00
generate_proposals_op_util_nms_test.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
given_tensor_byte_string_to_uint8_fill_op.cc Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385) 2018-08-15 14:26:50 -07:00
given_tensor_byte_string_to_uint8_fill_op.cu Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385) 2018-08-15 14:26:50 -07:00
given_tensor_byte_string_to_uint8_fill_op.h GCC-7 doesn't like the original syntax. (#10665) 2018-08-30 16:41:16 -07:00
given_tensor_fill_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
given_tensor_fill_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
given_tensor_fill_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
glu_op.cc
glu_op.cu
glu_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
group_norm_op.cc Optimize group_norm on cpu (#10246) 2018-08-06 14:26:09 -07:00
group_norm_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
group_norm_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
gru_unit_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
gru_unit_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
gru_unit_op_gpu.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
h_softmax_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
h_softmax_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
half_float_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
half_float_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
half_float_ops.h add a Float16UniformFill (#11123) 2018-09-04 23:28:22 -07:00
hard_sigmoid_op.cc Add CPU version of hard sigmoid operator to caffe2 (#10837) 2018-08-28 14:55:49 -07:00
hard_sigmoid_op.cu Add GPU version of HardSigmoid Op to Caffe2 (#10955) 2018-08-29 14:55:29 -07:00
hard_sigmoid_op.h Add CPU version of hard sigmoid operator to caffe2 (#10837) 2018-08-28 14:55:49 -07:00
heatmap_max_keypoint_op.cc oss heatmap_max_keypoint_op 2018-07-17 16:55:47 -07:00
heatmap_max_keypoint_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
if_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
if_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
if_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
im2col_op.cc bugfix im2col op 2018-03-30 21:00:44 -07:00
im2col_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
im2col_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
index_hash_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
index_hash_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
index_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
instance_norm_gradient_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
instance_norm_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
instance_norm_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
instance_norm_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
integral_image_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
integral_image_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
integral_image_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
is_empty_op.cc Remove BUILD_CAFFE2 and build everything (#8338) 2018-08-31 13:10:24 -07:00
is_empty_op.h Split IsEmptyOp (#10918) 2018-08-28 10:52:28 -07:00
jsd_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
jsd_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
key_split_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
key_split_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
last_n_window_collector.cc Cleaner semantics for Reserve (#10261) 2018-08-06 14:40:16 -07:00
layer_norm_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
layer_norm_op.cu Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
layer_norm_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
leaky_relu_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
leaky_relu_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
leaky_relu_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
lengths_pad_op.cc Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
lengths_pad_op.cu Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
lengths_pad_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
lengths_reducer_fused_8bit_rowwise_ops.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_reducer_fused_8bit_rowwise_ops.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_reducer_ops.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_reducer_ops.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_reducer_rowwise_8bit_ops.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_reducer_rowwise_8bit_ops.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
lengths_tile_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lengths_tile_op.h Make lengths_host_.CopyFrom synced in LengthsCosineCoherenceOp and LengthsTileOp (#10360) 2018-08-10 10:39:28 -07:00
lengths_tile_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lengths_top_k_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lengths_top_k_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
listwise_l2r_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
listwise_l2r_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
load_save_op.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
load_save_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
load_save_op_gpu.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
local_response_normalization_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
local_response_normalization_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
local_response_normalization_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
local_response_normalization_op_cudnn.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
locally_connected_op.cc Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
locally_connected_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
locally_connected_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
locally_connected_op_impl.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
locally_connected_op_util.cc Update Im2ColNd functions (#7505) 2018-05-12 15:59:50 -07:00
locally_connected_op_util.h Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
log_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
log_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
log_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
logit_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
logit_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
logit_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
loss_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
loss_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
loss_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lp_pool_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
lp_pool_op.cu Fix compile warnings 2018-09-01 21:41:25 -07:00
lpnorm_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
lpnorm_op.h [caffe2] Update ReduceOps (#6497) 2018-04-11 23:36:05 -07:00
lstm_unit_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
lstm_unit_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
lstm_unit_op_gpu.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
map_ops.cc Add interface to provide blob types to shape&type inference (#9643) 2018-07-24 11:58:05 -07:00
map_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
margin_ranking_criterion_op.cc Add documentation for margin arg in Caffe2 MarginRankingCriterionOp (#10186) 2018-08-03 11:45:51 -07:00
margin_ranking_criterion_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
margin_ranking_criterion_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
matmul_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
matmul_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
matmul_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
max_pool_with_index.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
max_pool_with_index_gpu.h HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
mean_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
mean_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mean_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mem_query_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
merge_id_lists_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
merge_id_lists_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
minmax_gradient_ops.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
minmax_ops.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
minmax_ops.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
mod_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
mod_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
moments_op.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
moments_op.cu Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
moments_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
multi_class_accuracy_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
multi_class_accuracy_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
multi_class_accuracy_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
negate_gradient_op.cc
negate_gradient_op.h
negate_gradient_op_gpu.cc
negative_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
negative_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
negative_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
ngram_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
ngram_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
no_default_engine_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
norm_planar_yuv_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
normalize_l1_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
normalize_l1_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
normalize_op.cc Keep kEps in one place to make sure they are consistent (#10334) 2018-08-09 08:27:42 -07:00
normalize_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
normalize_ops.cu Keep kEps in one place to make sure they are consistent (#10334) 2018-08-09 08:27:42 -07:00
numpy_tile_op.cc [caffe2][ONNX] Implement CPU NumpyTileOp and corresponding ONNX backend (#7053) 2018-04-27 19:58:15 -07:00
numpy_tile_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
one_hot_ops.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
one_hot_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
one_hot_ops.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
onnx_while_op.cc [caffe2] Add flag to ONNXWhile to skip scoping (#6910) 2018-04-24 16:53:22 -07:00
onnx_while_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
onnxifi_op.cc Make sure we don't relocate the weight name buffer (#10630) 2018-08-17 16:03:31 -07:00
onnxifi_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
op_utils_cudnn.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
operator_fallback_gpu.h caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
operator_fallback_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
order_switch_ops.cc generalize order switch ops for 1-3d (#10395) 2018-08-15 10:09:31 -07:00
order_switch_ops.cu generalize order switch ops for 1-3d (#10395) 2018-08-15 10:09:31 -07:00
order_switch_ops.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pack_rnn_sequence_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pack_rnn_sequence_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
pack_segments.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
pack_segments.cu Fix some more warnings (#11257) 2018-09-05 11:10:27 -07:00
pack_segments.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
pad_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
pad_op.h Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
pad_op_gpu.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
partition_ops.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
partition_ops.h s/DISABLE_COPY_AND_ASSIGN/AT_DISABLE_COPY_AND_ASSIGN/ (#10275) 2018-08-07 08:54:26 -07:00
percentile_op.cc Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
percentile_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
perplexity_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
perplexity_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
perplexity_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
piecewise_linear_transform_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
piecewise_linear_transform_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
piecewise_linear_transform_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
pool_gradient_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
pool_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
pool_op.cu Fix some more warnings (#11257) 2018-09-05 11:10:27 -07:00
pool_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pool_op_cudnn.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
pow_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
pow_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pow_op.h Fix app size check (#8256) 2018-06-07 15:34:22 -07:00
prefetch_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
prelu_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
prelu_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
prelu_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
prepend_dim_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
prepend_dim_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
prepend_dim_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
quant_decode_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
quant_decode_op.h codemod -d caffe2 --extensions cc,h CaffeTypeId TypeIdentifier (#10166) 2018-08-02 11:54:30 -07:00
rank_loss_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rank_loss_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
reciprocal_gradient_op.cc Adding reciprocal operator and a test 2018-07-27 18:24:43 -07:00
reciprocal_op.cc Adding reciprocal operator and a test 2018-07-27 18:24:43 -07:00
reciprocal_op.cu Adding reciprocal operator and a test 2018-07-27 18:24:43 -07:00
reciprocal_op.h Adding reciprocal operator and a test 2018-07-27 18:24:43 -07:00
reduce_ops.cc Add support for .norm() pytorch onnx export and ReduceL1/ReduceL2 caffe2 operators (#9299) 2018-07-14 10:54:13 -07:00
reduce_ops.cu Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072) 2018-07-01 00:25:34 -07:00
reduce_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
reducer_functors.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
reduction_front_back_ops.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
reduction_front_back_ops.cu Eanble python tests on ROCM (#9616) 2018-07-24 11:37:58 -07:00
reduction_front_back_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
reduction_ops.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
reduction_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
reduction_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
relu_n_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
relu_n_op.cu Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
relu_n_op.h Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
relu_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
relu_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
relu_op.h Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
relu_op_cudnn.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
remove_data_blocks_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
remove_data_blocks_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
replace_nan_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
replace_nan_op.cu
replace_nan_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
reservoir_sampling.cc Cleaner semantics for Reserve (#10261) 2018-08-06 14:40:16 -07:00
reshape_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
reshape_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
reshape_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
reshape_op_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
resize_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
resize_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
resize_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
reverse_packed_segs_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
reverse_packed_segs_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
reverse_packed_segs_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
rmac_regions_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
rmac_regions_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
rmac_regions_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_align_gradient_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_align_gradient_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_align_gradient_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_align_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_align_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_align_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_align_op_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
roi_align_rotated_gradient_op.cc Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
roi_align_rotated_gradient_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
roi_align_rotated_gradient_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_align_rotated_op.cc Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
roi_align_rotated_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
roi_align_rotated_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_pool_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_pool_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
roi_pool_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
rowmul_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rowmul_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
rsqrt_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
rsqrt_op.cu Add Cube and Cbrt Ops (#8991) 2018-06-28 14:55:30 -07:00
rsqrt_op.h Add Cube and Cbrt Ops (#8991) 2018-06-28 14:55:30 -07:00
scale_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
scale_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
scale_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
segment_reduction_op.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
segment_reduction_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
segment_reduction_op_gpu.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
selu_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
selu_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
selu_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
sequence_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
sequence_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
sequence_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
shape_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
shape_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
shape_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sigmoid_gradient_op.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
sigmoid_op.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
sigmoid_op.cu Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
sigmoid_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
sigmoid_op_cudnn.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
sin_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
sin_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
sin_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
sinh_op.cc Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
sinh_op.cu Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
sinh_op.h Implement Sinh and Cosh (#9213) 2018-07-10 18:55:31 -07:00
sinusoid_position_encoding_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sinusoid_position_encoding_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
slice_op.cc Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
slice_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
slice_op.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
softmax_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
softmax_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
softmax_op_cudnn.cc fix expty batch for softmax (#9075) 2018-07-01 16:40:14 -07:00
softmax_ops.cu Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
softmax_shared.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
softmax_shared.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
softmax_with_loss_op.cc Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
softmax_with_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
softplus_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
softplus_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
softplus_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
softsign_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
softsign_op.cu Fix some more compile warnings 2018-09-03 19:39:33 -07:00
softsign_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
space_batch_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
space_batch_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
space_batch_op_gpu.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
sparse_normalize_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
sparse_normalize_op.h Additional changes to make GPU builds work (#10507) 2018-08-16 13:25:27 -07:00
sparse_normalize_op_gpu.cu Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
sparse_to_dense_mask_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
sparse_to_dense_mask_op.h Move filler interface to operator schema (#10522) 2018-08-15 12:40:18 -07:00
sparse_to_dense_op.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
sparse_to_dense_op.cu HIP Operators Generator--> HipOpG (#9322) 2018-07-19 00:26:06 -07:00
sparse_to_dense_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
spatial_batch_norm_gradient_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
spatial_batch_norm_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
spatial_batch_norm_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
spatial_batch_norm_op_cudnn.cc empty batch for spatialBN (#8933) 2018-06-29 18:40:41 -07:00
spatial_softmax_with_loss_op.cc Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
spatial_softmax_with_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
sqr_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
sqr_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
sqr_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
sqrt_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
sqrt_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
sqrt_op_gpu.cc [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
square_root_divide_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
square_root_divide_op.h Optimize reduce ops for 2d and 3d (#9992) 2018-08-04 13:53:58 -07:00
stats_ops.cc Windows raw string fix (#10998) 2018-08-29 11:40:08 -07:00
stop_gradient.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
stop_gradient.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
stop_gradient_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
string_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
string_ops.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
string_ops_test.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
stump_func_op.cc [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
stump_func_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
stump_func_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
stylizer_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
summarize_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
summarize_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
summarize_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
swish_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
swish_op.cu [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
swish_op.h [Caffe2] Update elementwise ops to support numpy style boradcast (#8070) 2018-06-05 15:49:16 -07:00
tan_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
tan_op.cu Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
tan_op.h Update from Facebook (#8887) 2018-06-26 14:55:48 -07:00
tanh_gradient_op.cc Fix TanhGradientOperator linker errors (#10426) 2018-08-13 17:57:10 -07:00
tanh_op.cc Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
tanh_op.cu Move tanh function to math (#9328) 2018-07-11 13:59:50 -07:00
tanh_op.h Move tanh function to math (#9328) 2018-07-11 13:59:50 -07:00
tanh_op_cudnn.cc Add cudnn activation ops (#9379) 2018-07-12 23:18:56 -07:00
tensor_protos_db_input.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tensor_protos_db_input.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
tensor_protos_db_input_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
text_file_reader.cc Cleaner semantics for Reserve (#10261) 2018-08-06 14:40:16 -07:00
text_file_reader_utils.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
text_file_reader_utils.h Remove BUILD_CAFFE2 and build everything (#8338) 2018-08-31 13:10:24 -07:00
text_file_reader_utils_test.cc Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
thresholded_relu_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
thresholded_relu_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
thresholded_relu_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
tile_op.cc Updates to caffe2 operator documentation (#7917) 2018-05-29 14:38:56 -07:00
tile_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tile_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
top_k.cc Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946) 2018-08-28 11:57:08 -07:00
top_k.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
top_k.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
top_k_heap_selection.cuh Update from facebook (#6871) 2018-04-23 15:01:56 -07:00
top_k_radix_selection.cuh Fix gatherTopK template (#9231) 2018-07-18 08:25:23 -07:00
transpose_op.cc Update operator documentation with markdown descriptions and interfaces (#8085) 2018-06-15 19:02:24 -04:00
transpose_op.cu Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
transpose_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
transpose_op_cudnn.cc Add broadcast and reduce gradient (#6668) 2018-04-17 13:31:13 -07:00
tt_linear_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tt_linear_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
unique_ops.cc [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
unique_ops.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
unique_ops.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
upsample_op.cc bilinear upsample operator on CPU 2018-06-27 10:12:06 -07:00
upsample_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
utility_ops.cc Add cost inference to ConvGradient and WeightedSum operators (#10744) 2018-09-05 13:56:05 -07:00
utility_ops.cu Split gather op for easier smaller code size (#9916) 2018-07-27 17:15:33 -07:00
utility_ops.h Optimize WeightedSumOp for two inputs (#11049) 2018-09-01 11:54:55 -07:00
utility_ops_cudnn.cc Optimize WeightedSumOp for two inputs (#11049) 2018-09-01 11:54:55 -07:00
utility_ops_gpu_test.cc caffe2::DeviceType -> at::DeviceType (#11254) 2018-09-05 16:28:09 -07:00
utility_ops_test.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
variable_length_sequence_padding.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
variable_length_sequence_padding.h Fix Eigen issue on OS X with CUDA and nvcc compile (#9350) 2018-07-11 14:00:05 -07:00
weighted_multi_sampling_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
weighted_multi_sampling_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
weighted_sample_op.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
weighted_sample_op.cu Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
weighted_sample_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
while_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
while_op.h Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
while_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
workspace_ops.cc Remove template parameter from Tensor (#9939) 2018-07-27 10:56:39 -07:00
zero_gradient_op.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
zero_gradient_op.h Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
zero_gradient_op_gpu.cc Remove Apache headers from source. 2018-03-27 13:10:18 -07:00