pytorch/caffe2/operators
Rui Zhu 19fe2b9db4 Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18621

This diff added caffe2 support for onnxifi quantization.

Reviewed By: yinghai

Differential Revision: D14648767

fbshipit-source-id: 4ddb492cacbba6142305866e6dbb875880acaea3
2019-03-31 17:42:27 -07:00
..
experimental/c10/cpu Check kernel against function schema in c10 op registration (#18256) 2019-03-30 00:07:22 -07:00
hip Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
quantized Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
rnn refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
abs_op.cc
abs_op.cu
abs_op.h
accumulate_op.cc
accumulate_op.cu
accumulate_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
accuracy_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
accuracy_op.cu Tensor method rename ndim()->dim() - 1/3 (#16678) 2019-02-04 15:49:16 -08:00
accuracy_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
acos_op.cc
acos_op.cu
acos_op.h
activation_ops_cudnn.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
adjust_batch_op.cc Add AdjustBatch Op (#16676) 2019-02-06 19:15:41 -08:00
adjust_batch_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
affine_channel_op.cc Tensor construction codemod(ResizeLike) - 3/7 (#15122) 2018-12-14 02:08:37 -08:00
affine_channel_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
affine_channel_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
apmeter_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
apmeter_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
arg_ops.cc
arg_ops.cu
arg_ops.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
asin_op.cc
asin_op.cu
asin_op.h
assert_op.cc
assert_op.cu
assert_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
atan_op.cc
atan_op.cu
atan_op.h
atomic_ops.cc refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
batch_box_cox_op.cc Tensor construction codemod(ResizeLike) - 3/7 (#15122) 2018-12-14 02:08:37 -08:00
batch_box_cox_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
batch_bucketize_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
batch_bucketize_op.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
batch_gather_ops.cc Implements Gather operator for arbitrary axis, sharing the code with BatchGather. (#13756) 2018-12-04 11:54:28 -08:00
batch_gather_ops.cu Fix type-o in unsupported data type error message (#16537) 2019-02-05 10:46:46 -08:00
batch_gather_ops.h refactor caffe2 operator constructors - 1/9 (#17082) 2019-03-04 16:04:01 -08:00
batch_matmul_op.cc
batch_matmul_op.cu
batch_matmul_op.h refactor caffe2 operator constructors - 10/9 (#17659) 2019-03-06 15:11:47 -08:00
batch_matmul_op_gpu_test.cc
batch_matmul_op_test.cc
batch_moments_op.cc
batch_moments_op.cu
batch_moments_op.h refactor caffe2 operator constructors - 10/9 (#17659) 2019-03-06 15:11:47 -08:00
batch_sparse_to_dense_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
batch_sparse_to_dense_op.h refactor caffe2 operator constructors - 10/9 (#17659) 2019-03-06 15:11:47 -08:00
bbox_transform_op.cc Removed obsolete argument correct_transform_coords in bbox_transform op. (#16723) 2019-02-20 13:22:33 -08:00
bbox_transform_op.h Improve caffe2 operator wrapping (#17743) 2019-03-14 08:54:15 -07:00
bisect_percentile_op.cc
bisect_percentile_op.h refactor caffe2 operator constructors - 10/9 (#17659) 2019-03-06 15:11:47 -08:00
boolean_mask_ops.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
boolean_mask_ops.cu Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
boolean_mask_ops.h refactor caffe2 operator constructors - 10/9 (#17659) 2019-03-06 15:11:47 -08:00
boolean_unmask_ops.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
boolean_unmask_ops.cu Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
boolean_unmask_ops.h
boolean_unmask_ops_test.cc
box_with_nms_limit_op.cc register BoxWithNMSLimit with C10 2019-03-29 13:41:40 -07:00
box_with_nms_limit_op.h register BoxWithNMSLimit with C10 2019-03-29 13:41:40 -07:00
byte_weight_dequant_op.cc
byte_weight_dequant_op.h Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
cast_op.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
cast_op.cu Tensor construction codemod (#16568) 2019-02-05 18:51:02 -08:00
cast_op.h refactor caffe2 operator constructors - 11/9 (#17722) 2019-03-08 12:38:54 -08:00
cbrt_op.cc
cbrt_op.cu
cbrt_op.h
ceil_op.cc
ceil_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
ceil_op.h Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
channel_backprop_stats_op.cc
channel_backprop_stats_op.cu Tensor method rename ndim()->dim() - 1/3 (#16678) 2019-02-04 15:49:16 -08:00
channel_backprop_stats_op.h refactor caffe2 operator constructors - 11/9 (#17722) 2019-03-08 12:38:54 -08:00
channel_shuffle_op.cc Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
channel_shuffle_op.cu Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
channel_shuffle_op.h refactor caffe2 operator constructors - 11/9 (#17722) 2019-03-08 12:38:54 -08:00
channel_stats_op.cc Optimize channel_stats_op (#16243) 2019-03-12 12:08:00 -07:00
channel_stats_op.cu Optimize channel_stats_op (#16243) 2019-03-12 12:08:00 -07:00
channel_stats_op.h Optimize channel_stats_op (#16243) 2019-03-12 12:08:00 -07:00
clip_op.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
clip_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
clip_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
CMakeLists.txt Add functor- and function-based kernel registration API (#18162) 2019-03-30 00:07:19 -07:00
collect_and_distribute_fpn_rpn_proposals_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
collect_and_distribute_fpn_rpn_proposals_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
communicator_op.cc
communicator_op_gpu.cc
concat_split_op.cc add tensor and cost inference functions (#17684) 2019-03-06 23:34:02 -08:00
concat_split_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
concat_split_op_gpu.cc
conditional_op.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
conditional_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
conv_gradient_op.cc
conv_op.cc
conv_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
conv_op_cache_cudnn.cc
conv_op_cache_cudnn.h Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
conv_op_cache_cudnn_test.cc
conv_op_cudnn.cc Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
conv_op_eigen.cc refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
conv_op_gpu.cc
conv_op_impl.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
conv_op_shared.cc
conv_op_shared.h
conv_op_shared_gpu.cc
conv_pool_op_base.h Replace the remaining usages of IntList in caffe2 to IntArrayRef 2019-03-21 16:34:38 -07:00
conv_transpose_gradient_op.cc
conv_transpose_op.cc
conv_transpose_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
conv_transpose_op_cudnn.cc refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
conv_transpose_op_gpu.cc
conv_transpose_op_impl.h Change SetOutputSize in ConvTransposeUnpoolBaseOp (#16179) 2019-01-28 18:28:11 -08:00
conv_transpose_op_mobile.cc Make C10_MOBILE consistent with how feature macros are usually used (#17481) 2019-02-27 17:57:51 -08:00
conv_transpose_op_mobile.h Make C10_MOBILE consistent with how feature macros are usually used (#17481) 2019-02-27 17:57:51 -08:00
conv_transpose_op_mobile_impl.h Make C10_MOBILE consistent with how feature macros are usually used (#17481) 2019-02-27 17:57:51 -08:00
conv_transpose_op_mobile_test.cc
conv_transpose_unpool_op_base.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
copy_op.cc
copy_op.cu Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
copy_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
cos_op.cc
cos_op.cu
cos_op.h
cosh_op.cc
cosh_op.cu
cosh_op.h
cosine_embedding_criterion_op.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
cosine_embedding_criterion_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
cosine_embedding_criterion_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
counter_ops.cc Apply modernize-use-override - 2/2 2019-02-13 21:01:28 -08:00
counter_ops.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
counter_ops_gpu.cc
crash_op.cc Caffe2: crash op (#18207) 2019-03-22 11:52:01 -07:00
create_scope_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
create_scope_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
crf_viterbi_op.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
cross_entropy_op.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
cross_entropy_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
cross_entropy_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
ctc_beam_search_decoder_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
ctc_beam_search_decoder_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
ctc_greedy_decoder_op.cc fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
ctc_greedy_decoder_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
cube_op.cc
cube_op.cu
cube_op.h
data_couple.cc
data_couple.h
data_couple_gpu.cu
dataset_ops.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
dataset_ops.h
deform_conv_gradient_op.cc
deform_conv_op.cc
deform_conv_op.cu Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
deform_conv_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
deform_conv_op_impl.h Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
dense_vector_to_id_list_op.cc add dense vector to id_list operator (#15090) 2018-12-18 16:27:38 -08:00
dense_vector_to_id_list_op.h Tensor construction codemod (#16568) 2019-02-05 18:51:02 -08:00
depthwise_3x3_conv_op_cudnn.cu Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
distance_op.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
distance_op.cu Tensor construction codemod (#16568) 2019-02-05 18:51:02 -08:00
distance_op.h Move math::Axpy function to elementwise lib (#18316) 2019-03-26 12:19:19 -07:00
do_op.cc
do_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
do_op_gpu.cc
dropout_op.cc Tensor construction: combine Resize+mutable_data - 1/4 (#13942) 2018-11-19 15:33:50 -08:00
dropout_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
dropout_op.h refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
dropout_op_cudnn.cc refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
elementwise_add_gradient_op.cc
elementwise_add_op.cc
elementwise_add_op.h Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
elementwise_add_op_gpu.cc
elementwise_div_gradient_op.cc refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
elementwise_div_op.cc
elementwise_div_op.cu Remaining Tensor API fixes - dims() -> sizes() (#15743) 2019-01-15 18:42:02 -08:00
elementwise_div_op.h
elementwise_linear_op.cc Export ElementwiseLinear to ONNX (Mul + Add). (#17411) 2019-02-25 08:11:13 -08:00
elementwise_linear_op.cu Tensor method rename ndim()->dim() - 2/3 (#16679) 2019-02-04 11:12:57 -08:00
elementwise_linear_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
elementwise_logical_ops.cc
elementwise_logical_ops.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
elementwise_mul_gradient_op.cc
elementwise_mul_op.cc
elementwise_mul_op.cu
elementwise_mul_op.h
elementwise_op_gpu_test.cc
elementwise_op_test.cc
elementwise_op_test.h
elementwise_ops.cc Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
elementwise_ops.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
elementwise_ops.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
elementwise_ops_schema.cc Fix typo in Github links in elementwise_ops_schema.cc (#18018) 2019-03-26 15:37:26 -07:00
elementwise_ops_utils.cc Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
elementwise_ops_utils.h Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
elementwise_sub_gradient_op.cc
elementwise_sub_op.cc
elementwise_sub_op.h Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
elementwise_sub_op_gpu.cc
elementwise_sum_op.cc add tensor and cost inference functions (#17684) 2019-03-06 23:34:02 -08:00
elu_op.cc
elu_op.cu
elu_op.h
elu_op_cudnn.cc refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
enforce_finite_op.cc
enforce_finite_op.cu
enforce_finite_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
ensure_clipped_op.cc
ensure_clipped_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
ensure_cpu_output_op.cc
ensure_cpu_output_op.cu
ensure_cpu_output_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
erf_op.cc Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator 2019-01-17 09:18:08 -08:00
erf_op.cu Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator 2019-01-17 09:18:08 -08:00
erf_op.h Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator 2019-01-17 09:18:08 -08:00
exp_op.cc
exp_op.h
exp_op_gpu.cc
expand_op.cc
expand_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
expand_op_gpu.cc
expand_squeeze_dims_op.cc
expand_squeeze_dims_op.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
expand_squeeze_dims_op_gpu.cc
fc_inference.cc remove redundant inference functions for FC (#17407) 2019-02-22 20:13:20 -08:00
fc_inference.h remove redundant inference functions for FC (#17407) 2019-02-22 20:13:20 -08:00
feature_maps_ops.cc
feature_maps_ops.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
feed_blob_op.cc
feed_blob_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
filler_op.cc
filler_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
filler_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
find_duplicate_elements_op.cc
find_duplicate_elements_op.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
find_op.cc
find_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
find_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
flatten_op.cc add tensor and cost inference functions (#17684) 2019-03-06 23:34:02 -08:00
flatten_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
flexible_top_k.cc Tensor construction: combine Resize+mutable_data - 2/4 (#14205) 2018-11-30 10:46:58 -08:00
flexible_top_k.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
floor_op.cc
floor_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
floor_op.h Tensor construction codemod(ResizeLike) - 4/7 (#15088) 2018-12-13 13:39:56 -08:00
free_op.cc
free_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
free_op_gpu.cc
fully_connected_op.cc Typo fix (#18089) 2019-03-16 15:07:01 -07:00
fully_connected_op.h Int8 FC performance debugging (#17700) 2019-03-08 19:03:54 -08:00
fully_connected_op_gpu.cc Integrate rocBLAS fp16 api into Caffe2 (#14882) 2018-12-10 17:54:06 -08:00
fused_rowwise_8bit_conversion_ops.cc move half<->float conversions to oss operators (#17548) 2019-03-07 13:00:13 -08:00
fused_rowwise_8bit_conversion_ops.h change the epsilon for fp32/fp16 to uint8 to be the same (#17062) 2019-02-15 18:33:37 -08:00
fused_rowwise_random_quantization_ops.cc minimize header file includes from _avx2.cc (#14950) 2018-12-13 00:18:11 -08:00
fused_rowwise_random_quantization_ops.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
gather_fused_8bit_rowwise_op.cc
gather_fused_8bit_rowwise_op.h Tensor construction: combine Resize+mutable_data - 2/4 (#14205) 2018-11-30 10:46:58 -08:00
gather_op.cc Implements Gather operator for arbitrary axis, sharing the code with BatchGather. (#13756) 2018-12-04 11:54:28 -08:00
gather_op.cu Implements Gather operator for arbitrary axis, sharing the code with BatchGather. (#13756) 2018-12-04 11:54:28 -08:00
gather_op.cuh Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
gather_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
gather_ranges_to_dense_op.cc
gather_ranges_to_dense_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
generate_proposals_op.cc Removed obsolete argument correct_transform_coords in bbox_transform op. (#16723) 2019-02-20 13:22:33 -08:00
generate_proposals_op.cu Expose cuda kernel for caffe2::GenerateProposals 2019-03-04 14:59:08 -08:00
generate_proposals_op.h Expose cuda kernel for caffe2::GenerateProposals 2019-03-04 14:59:08 -08:00
generate_proposals_op_gpu_test.cc Revert D14446895: [C2] Implement rotated generate_proposals_op without opencv dependency (~2x faster) 2019-03-23 09:38:55 -07:00
generate_proposals_op_test.cc Implement rotated generate_proposals_op without opencv dependency (CPU version) 2019-03-28 17:02:50 -07:00
generate_proposals_op_util_boxes.h Removed obsolete argument correct_transform_coords in bbox_transform op. (#16723) 2019-02-20 13:22:33 -08:00
generate_proposals_op_util_boxes_test.cc Removed obsolete argument correct_transform_coords in bbox_transform op. (#16723) 2019-02-20 13:22:33 -08:00
generate_proposals_op_util_nms.h Implement rotated generate_proposals_op without opencv dependency (CPU version) 2019-03-28 17:02:50 -07:00
generate_proposals_op_util_nms_gpu.cu Rotated boxes support for GPU GenerateProposals op (#15470) 2018-12-26 18:03:56 -08:00
generate_proposals_op_util_nms_gpu.h Rotated boxes support for GPU GenerateProposals op (#15470) 2018-12-26 18:03:56 -08:00
generate_proposals_op_util_nms_gpu_test.cc Implement rotated generate_proposals_op without opencv dependency (CPU version) 2019-03-28 17:02:50 -07:00
generate_proposals_op_util_nms_test.cc Implement rotated generate_proposals_op without opencv dependency (CPU version) 2019-03-28 17:02:50 -07:00
given_tensor_byte_string_to_uint8_fill_op.cc
given_tensor_byte_string_to_uint8_fill_op.cu
given_tensor_byte_string_to_uint8_fill_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
given_tensor_fill_op.cc
given_tensor_fill_op.cu
given_tensor_fill_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
glu_op.cc
glu_op.cu
glu_op.h refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08:00
group_norm_op.cc Optimize group_norm_op (#17945) 2019-03-21 13:05:01 -07:00
group_norm_op.cu Optimize group_norm_op (#17945) 2019-03-21 13:05:01 -07:00
group_norm_op.h Optimize group_norm_op (#17945) 2019-03-21 13:05:01 -07:00
gru_unit_op.cc
gru_unit_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
gru_unit_op_gpu.cu
h_softmax_op.cc Tensor reinitialization codemod - 3/5 (#15912) 2019-01-16 19:49:01 -08:00
h_softmax_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
half_float_ops.cc Fix half-float conversion ops to handle tensors larger than 2B of params (#17952) 2019-03-12 23:03:22 -07:00
half_float_ops.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
half_float_ops.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
half_float_ops_test.cc move half<->float conversions to oss operators (#17548) 2019-03-07 13:00:13 -08:00
hard_sigmoid_op.cc
hard_sigmoid_op.cu
hard_sigmoid_op.h
heatmap_max_keypoint_op.cc Revert D13866214: [c10] Expose HeatmapMaxKeypoints to torch 2019-02-06 08:26:16 -08:00
heatmap_max_keypoint_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
if_op.cc
if_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
if_op_gpu.cc
im2col_op.cc
im2col_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
im2col_op_gpu.cc
index_hash_ops.cc
index_hash_ops.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
index_ops.cc refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
index_ops.h move structs to header file (#14728) 2018-12-04 00:42:41 -08:00
inference_lstm_op.cc Use SetOutputTensor instead of copying outputs manually (#17770) 2019-03-28 16:01:33 -07:00
inference_lstm_op.h Implement a Caffe2 standalone LSTM operator (#17726) 2019-03-07 01:08:49 -08:00
instance_norm_gradient_op.cc Tensor reinitialization codemod - 3/5 (#15912) 2019-01-16 19:49:01 -08:00
instance_norm_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
instance_norm_op.cu Tensor method rename ndim()->dim() - 2/3 (#16679) 2019-02-04 11:12:57 -08:00
instance_norm_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
integral_image_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
integral_image_op.cu Tensor method rename ndim()->dim() - 2/3 (#16679) 2019-02-04 11:12:57 -08:00
integral_image_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
is_empty_op.cc
is_empty_op.h Tensor construction: combine Resize+mutable_data - 2/4 (#14205) 2018-11-30 10:46:58 -08:00
jsd_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
jsd_op.h
key_split_ops.cc
key_split_ops.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
last_n_window_collector.cc refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
layer_norm_op.cc Expose c10 operators to caffe2 by operator name (#18160) 2019-03-26 12:36:11 -07:00
layer_norm_op.cu Expose c10 operators to caffe2 by operator name (#18160) 2019-03-26 12:36:11 -07:00
layer_norm_op.h Expose c10 operators to caffe2 by operator name (#18160) 2019-03-26 12:36:11 -07:00
leaky_relu_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
leaky_relu_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
leaky_relu_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
length_split_op.cc
length_split_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
lengths_pad_op.cc
lengths_pad_op.cu
lengths_pad_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
lengths_reducer_fused_8bit_rowwise_ops.cc Add filler for SparseLengthsWeightedSum (#13949) 2018-11-16 11:31:05 -08:00
lengths_reducer_fused_8bit_rowwise_ops.h Tensor construction: combine Resize+mutable_data - 2/4 (#14205) 2018-11-30 10:46:58 -08:00
lengths_reducer_ops.cc Add onnxifi support to SparseLengthsWeightedSum (#14210) 2018-11-21 15:47:24 -08:00
lengths_reducer_ops.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
lengths_reducer_rowwise_8bit_ops.cc
lengths_reducer_rowwise_8bit_ops.h Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
lengths_tile_op.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
lengths_tile_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
lengths_tile_op.h Tensor reinitialization codemod - 3/5 (#15912) 2019-01-16 19:49:01 -08:00
lengths_top_k_op.cc Tensor construction: combine Resize+mutable_data - 2/4 (#14205) 2018-11-30 10:46:58 -08:00
lengths_top_k_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
listwise_l2r_op.cc Tensor reinitialization codemod - 3/5 (#15912) 2019-01-16 19:49:01 -08:00
listwise_l2r_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
load_save_op.cc
load_save_op.h Change message on unknown db type to be friendly (#17795) 2019-03-08 10:46:24 -08:00
load_save_op_gpu.cc
local_response_normalization_op.cc Move math::Axpy function to elementwise lib (#18316) 2019-03-26 12:19:19 -07:00
local_response_normalization_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
local_response_normalization_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
local_response_normalization_op_cudnn.cc refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
locally_connected_op.cc
locally_connected_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
locally_connected_op_gpu.cc
locally_connected_op_impl.h Back out "[pt1][tensor] Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize" (#16516) 2019-01-30 12:50:38 -08:00
locally_connected_op_util.cc
locally_connected_op_util.h
log_op.cc Revert D10439558: Add cost for non-linear ops 2018-11-16 23:30:05 -08:00
log_op.h
log_op_gpu.cc
logit_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
logit_op.cu
logit_op.h refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
loss_op.cc
loss_op.cu
loss_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
lp_pool_op.cc Back out "[pt1][tensor] Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize" (#16516) 2019-01-30 12:50:38 -08:00
lp_pool_op.cu Tensor method rename size()->numel() - 1/3 2019-02-04 23:33:17 -08:00
lpnorm_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
lpnorm_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
lstm_unit_op.cc Revert D10439558: Add cost for non-linear ops 2018-11-16 23:30:05 -08:00
lstm_unit_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
lstm_unit_op_gpu.cu
lstm_utils.h Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
map_ops.cc
map_ops.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
margin_ranking_criterion_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
margin_ranking_criterion_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
margin_ranking_criterion_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
matmul_op.cc
matmul_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
matmul_op_gpu.cc
max_pool_with_index.cu Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
max_pool_with_index_gpu.h
mean_op.cc
mean_op.cu
mean_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
mem_query_op.cu Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
merge_id_lists_op.cc
merge_id_lists_op.h Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
minmax_gradient_ops.cc Separate elementwise level2 math functions (#16753) 2019-02-07 18:38:26 -08:00
minmax_ops.cc Separate elementwise level2 math functions (#16753) 2019-02-07 18:38:26 -08:00
minmax_ops.cu Separate elementwise level2 math functions (#16753) 2019-02-07 18:38:26 -08:00
minmax_ops.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
mod_op.cc Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
mod_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
moments_op.cc
moments_op.cu
moments_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
multi_class_accuracy_op.cc Tensor construction: combine Resize+mutable_data - 3/4 (#13944) 2018-11-19 15:28:13 -08:00
multi_class_accuracy_op.cu Tensor method rename ndim()->dim() - 2/3 (#16679) 2019-02-04 11:12:57 -08:00
multi_class_accuracy_op.h
negate_gradient_op.cc
negate_gradient_op.h Shut up "address will always evaluate to 'true'" warnings (#14774) 2018-12-05 21:18:31 -08:00
negate_gradient_op_gpu.cc
negative_op.cc
negative_op.h
negative_op_gpu.cc
ngram_ops.cc
ngram_ops.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
no_default_engine_op.h
norm_planar_yuv_op.cc Apply modernize-use-override - 2/2 2019-02-13 21:01:28 -08:00
normalize_l1_op.cc
normalize_l1_op.h Tensor construction codemod(ResizeLike) - 5/7 (#15084) 2018-12-13 12:42:52 -08:00
normalize_op.cc Fix compilation error (#17860) 2019-03-11 10:26:42 -07:00
normalize_op.h Fix compilation error (#17860) 2019-03-11 10:26:42 -07:00
normalize_ops.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
numpy_tile_op.cc
numpy_tile_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
one_hot_ops.cc Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
one_hot_ops.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
one_hot_ops.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
onnx_while_op.cc
onnx_while_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
onnxifi_op.cc Adding quantized tensor shape/type info support for caffe2=>glow in caffe2 side (#18621) 2019-03-31 17:42:27 -07:00
onnxifi_op.h Move weight offload inside backend construction functor (#18385) 2019-03-26 21:03:17 -07:00
op_utils_cudnn.h
operator_fallback_gpu.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
operator_fallback_gpu_test.cc Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
order_switch_ops.cc
order_switch_ops.h Tensor construction: combine Resize+mutable_data - 3/4 (#13944) 2018-11-19 15:28:13 -08:00
order_switch_ops_cudnn.cc Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
order_switch_ops_gpu.cc
pack_rnn_sequence_op.cc
pack_rnn_sequence_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
pack_segments.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
pack_segments.cu Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
pack_segments.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
pad_op.cc Back out "[pt1][tensor] Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize" (#16516) 2019-01-30 12:50:38 -08:00
pad_op.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
pad_op_gpu.cu Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
partition_ops.cc
partition_ops.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
percentile_op.cc Tensor reinitialization codemod - 3/5 (#15912) 2019-01-16 19:49:01 -08:00
percentile_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
perplexity_op.cc Tensor construction: combine Resize+mutable_data - 3/4 (#13944) 2018-11-19 15:28:13 -08:00
perplexity_op.cu Tensor construction codemod (#16568) 2019-02-05 18:51:02 -08:00
perplexity_op.h
piecewise_linear_transform_op.cc
piecewise_linear_transform_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
piecewise_linear_transform_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
pool_gradient_op.cc Add count_include_pad to average_pool_gradient_op (#15997) 2019-01-15 16:56:40 -08:00
pool_op.cc Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
pool_op.cu Separate reduce functions from math (#16929) 2019-02-13 17:50:47 -08:00
pool_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
pool_op_cudnn.cc Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08:00
pool_op_util.cc Add count_include_pad arg for PoolOpGradient on CPU and fix ARM performance issue. (#15651) 2019-01-03 00:18:47 -08:00
pool_op_util.h Add count_include_pad arg for PoolOpGradient on CPU and fix ARM performance issue. (#15651) 2019-01-03 00:18:47 -08:00
pow_op.cc Revert D10439558: Add cost for non-linear ops 2018-11-16 23:30:05 -08:00
pow_op.cu
pow_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
prefetch_op.h Replace SwitchToDevice(0) with SwitchToDevice() (#15126) 2018-12-17 15:15:00 -08:00
prelu_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
prelu_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
prelu_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
prepend_dim_op.cc Support conversion from Caffe2 MergeDim to ONNX Reshape + Squeeze. (#16189) 2019-02-13 15:53:38 -08:00
prepend_dim_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
prepend_dim_op_gpu.cc
quant_decode_op.cc
quant_decode_op.h refactor caffe2 operator constructors - 6/9 (#17087) 2019-02-28 14:23:57 -08:00
rank_loss_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
rank_loss_op.h
reciprocal_gradient_op.cc
reciprocal_op.cc
reciprocal_op.cu
reciprocal_op.h
reduce_front_back_max_ops.cc
reduce_front_back_max_ops.cu
reduce_front_back_max_ops.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
reduce_front_back_mean_ops.cc Export ReduceMean/ReduceFrontMean/ReduceBackMean (Caffe2) to ReduceMean (ONNX). (#16727) 2019-02-12 13:35:32 -08:00
reduce_front_back_sum_mean_ops.cu
reduce_front_back_sum_mean_ops.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
reduce_front_back_sum_ops.cc
reduce_ops.cc Export ReduceMean/ReduceFrontMean/ReduceBackMean (Caffe2) to ReduceMean (ONNX). (#16727) 2019-02-12 13:35:32 -08:00
reduce_ops.cu
reduce_ops.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
reducer_functors.h Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
reduction_ops.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
reduction_ops.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
reduction_ops.h Reapply D14078519 (#17596) 2019-03-06 13:51:00 -08:00
relu_n_op.cc
relu_n_op.cu
relu_n_op.h
relu_op.cc
relu_op.cu Optimize relu op on GPU (#18506) 2019-03-29 00:23:24 -07:00
relu_op.h
remove_data_blocks_op.cc
remove_data_blocks_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
replace_nan_op.cc
replace_nan_op.cu
replace_nan_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
reservoir_sampling.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
reshape_op.cc
reshape_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
reshape_op_gpu.cc
reshape_op_gpu_test.cc
resize_op.cc Add NHWC support to Resize Operator (#15553) 2019-01-08 16:44:17 -08:00
resize_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
resize_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
reverse_packed_segs_op.cc
reverse_packed_segs_op.cu Tensor method rename ndim()->dim() - 2/3 (#16679) 2019-02-04 11:12:57 -08:00
reverse_packed_segs_op.h Tensor construction: combine Resize+mutable_data - 3/4 (#13944) 2018-11-19 15:28:13 -08:00
rmac_regions_op.cc Remove Context dependency from Tensor class (#14269) 2018-11-28 15:45:38 -08:00
rmac_regions_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
rmac_regions_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
roi_align_gradient_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
roi_align_gradient_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
roi_align_gradient_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
roi_align_op.cc register RoIAlign with C10 2019-03-14 11:55:29 -07:00
roi_align_op.cu register RoIAlign with C10 2019-03-14 11:55:29 -07:00
roi_align_op.h register RoIAlign with C10 2019-03-14 11:55:29 -07:00
roi_align_op_gpu_test.cc
roi_align_rotated_gradient_op.cc
roi_align_rotated_gradient_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
roi_align_rotated_gradient_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
roi_align_rotated_op.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
roi_align_rotated_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
roi_align_rotated_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
roi_pool_op.cc
roi_pool_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
roi_pool_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
rowmul_op.cc
rowmul_op.h Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
rsqrt_op.cc
rsqrt_op.cu Separate elementwise level2 math functions (#16753) 2019-02-07 18:38:26 -08:00
rsqrt_op.h
scale_op.cc
scale_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
scale_op_gpu.cc
segment_reduction_op.cc Windows shared build (#13550) 2018-11-16 12:16:28 -08:00
segment_reduction_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
segment_reduction_op_gpu.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
selu_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
selu_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
selu_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
sequence_ops.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
sequence_ops.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
sequence_ops.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
shape_op.cc
shape_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
shape_op_gpu.cc
sigmoid_gradient_op.cc
sigmoid_op.cc Revert D10439558: Add cost for non-linear ops 2018-11-16 23:30:05 -08:00
sigmoid_op.cu
sigmoid_op.h
sigmoid_op_cudnn.cc
sin_op.cc
sin_op.cu
sin_op.h
sinh_op.cc
sinh_op.cu
sinh_op.h
sinusoid_position_encoding_op.cc
sinusoid_position_encoding_op.h refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
slice_op.cc
slice_op.cu refactor caffe2 operator constructors - 7/9 (#17088) 2019-02-28 14:23:53 -08:00
slice_op.h fix -Wsign-compare warnings for some files inside c2 (#18123) 2019-03-19 10:39:20 -07:00
softmax_op.cc Fix SoftmaxOps (#16049) 2019-01-18 12:30:59 -08:00
softmax_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
softmax_op_cudnn.cc refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
softmax_ops.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
softmax_shared.cc
softmax_shared.h
softmax_with_loss_op.cc Fix SoftmaxOps (#16049) 2019-01-18 12:30:59 -08:00
softmax_with_loss_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
softplus_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
softplus_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
softplus_op.h
softsign_op.cc
softsign_op.cu
softsign_op.h
space_batch_op.cc
space_batch_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
space_batch_op_gpu.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
sparse_normalize_op.cc
sparse_normalize_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
sparse_normalize_op_gpu.cu
sparse_to_dense_mask_op.cc
sparse_to_dense_mask_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
sparse_to_dense_op.cc
sparse_to_dense_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
sparse_to_dense_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
spatial_batch_norm_gradient_op.cc Optimize SpatialBNOp on GPU (#16395) 2019-01-28 09:36:45 -08:00
spatial_batch_norm_op.cc Revert D13747581: Optimize SpatialBN on GPU 2019-01-24 15:26:37 -08:00
spatial_batch_norm_op.cu Optimize SpatialBNOp on GPU (#16395) 2019-01-28 09:36:45 -08:00
spatial_batch_norm_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
spatial_batch_norm_op_cudnn.cu Optimize SpatialBNOp on GPU (#16395) 2019-01-28 09:36:45 -08:00
spatial_batch_norm_op_impl.cuh Separate gpu reduce functions (#17146) 2019-02-20 14:49:01 -08:00
spatial_softmax_with_loss_op.cc Fix SoftmaxOps (#16049) 2019-01-18 12:30:59 -08:00
spatial_softmax_with_loss_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
sqr_op.cc
sqr_op.h
sqr_op_gpu.cc
sqrt_op.cc
sqrt_op.h Seperate level1 elementwise functions from math (#16397) 2019-01-30 00:04:12 -08:00
sqrt_op_gpu.cc
square_root_divide_op.cc
square_root_divide_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
stats_ops.cc refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
stats_put_ops.cc
stats_put_ops.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
stop_gradient.cc
stop_gradient.h Remove Context dependency from Tensor class (#14269) 2018-11-28 15:45:38 -08:00
stop_gradient_gpu.cc
string_ops.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
string_ops.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
string_ops_test.cc Remove caffe2::ShareData (#16139) 2019-01-25 15:39:11 -08:00
stump_func_op.cc Tensor construction codemod(ResizeLike) - 6/7 (#15137) 2018-12-13 12:47:33 -08:00
stump_func_op.cu Tensor method rename size()->numel() - 2/3 (#16745) 2019-02-04 23:59:10 -08:00
stump_func_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
stylizer_ops.cc refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
summarize_op.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
summarize_op.cu Tensor method rename size()->numel() - 3/3 (#16747) 2019-02-04 23:54:33 -08:00
summarize_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
swish_op.cc Tensor construction codemod(ResizeLike) - 7/7 (#15087) 2018-12-20 15:33:07 -08:00
swish_op.cu
swish_op.h
tan_op.cc
tan_op.cu
tan_op.h
tanh_gradient_op.cc
tanh_op.cc Revert D10439558: Add cost for non-linear ops 2018-11-16 23:30:05 -08:00
tanh_op.cu
tanh_op.h
tanh_op_cudnn.cc
tensor_protos_db_input.cc
tensor_protos_db_input.h Remove partially initialized Tensor in Deserialization (#14197) 2018-12-10 17:17:29 -08:00
tensor_protos_db_input_gpu.cc
text_file_reader.cc Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
text_file_reader_utils.cc
text_file_reader_utils.h
text_file_reader_utils_test.cc Apply modernize-use-override - 2/2 2019-02-13 21:01:28 -08:00
thresholded_relu_op.cc Tensor construction codemod(ResizeLike) - 7/7 (#15087) 2018-12-20 15:33:07 -08:00
thresholded_relu_op.cu Tensor method rename size()->numel() - 3/3 (#16747) 2019-02-04 23:54:33 -08:00
thresholded_relu_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
tile_op.cc Temp fix for TileOp backward compatibility (#18026) 2019-03-14 13:54:29 -07:00
tile_op.cu Optimize TileOp (#17290) 2019-03-12 12:16:30 -07:00
tile_op.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
top_k.cc Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
top_k.cu Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
top_k.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
top_k_heap_selection.cuh Do not apply these explicit unroll pragmas for ROCm. (#18204) 2019-03-20 08:06:07 -07:00
top_k_radix_selection.cuh Separate affine_channel from math and optimize it (#16135) 2019-01-18 22:40:16 -08:00
transpose_op.cc
transpose_op.cu Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
transpose_op.h Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
transpose_op_cudnn.cc Update math::Transpose to support tensor with size > 2G (#17670) 2019-03-20 18:22:21 -07:00
tt_linear_op.cc
tt_linear_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
unique_ops.cc Tensor construction codemod(ResizeLike) - 7/7 (#15087) 2018-12-20 15:33:07 -08:00
unique_ops.cu Tensor construction codemod (#16568) 2019-02-05 18:51:02 -08:00
unique_ops.h Tensor reinitialization codemod - 4/5 (#15967) 2019-01-11 16:41:19 -08:00
upsample_op.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
upsample_op.cu Tensor method rename size()->numel() - 3/3 (#16747) 2019-02-04 23:54:33 -08:00
upsample_op.h refactor caffe2 operator constructors - 8/9 (#17089) 2019-02-28 14:45:20 -08:00
utility_ops.cc ONNX Export IsNan op 2019-03-15 12:19:03 -07:00
utility_ops.cu Separate elementwise level2 math functions (#16753) 2019-02-07 18:38:26 -08:00
utility_ops.h Tensor construction codemod(raw_mutable_data) (#16373) 2019-03-29 18:36:46 -07:00
utility_ops_gpu_test.cc
utility_ops_test.cc
variable_length_sequence_padding.cc
variable_length_sequence_padding.h refactor caffe2 operator constructors - 9/9 (#17090) 2019-02-28 09:53:18 -08:00
weighted_multi_sampling_op.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
weighted_multi_sampling_op.h refactor caffe2 operator constructors - 9/9 (#17090) 2019-02-28 09:53:18 -08:00
weighted_sample_op.cc Tensor construction: combine Resize+mutable_data - 4/4 (#13856) 2018-11-27 12:34:25 -08:00
weighted_sample_op.cu Tensor reinitialization codemod - 4/5 (#15967) 2019-01-11 16:41:19 -08:00
weighted_sample_op.h refactor caffe2 operator constructors - 9/9 (#17090) 2019-02-28 09:53:18 -08:00
while_op.cc
while_op.h refactor caffe2 operator constructors - 9/9 (#17090) 2019-02-28 09:53:18 -08:00
while_op_gpu.cc
workspace_ops.cc refactor caffe2 operator constructors - 9/9 (#17090) 2019-02-28 09:53:18 -08:00
zero_gradient_op.cc
zero_gradient_op.h
zero_gradient_op_gpu.cc