tree: 52bc4182c9b1f74e3088030381469199c95ab00e [path history] [tgz]
  1. accumulate_op.cc
  2. accumulate_op.cu
  3. accumulate_op.h
  4. accuracy_op.cc
  5. accuracy_op.cu
  6. accuracy_op.h
  7. apmeter_op.cc
  8. apmeter_op.h
  9. atomic_ops.cc
  10. batch_box_cox_op.cc
  11. batch_box_cox_op.h
  12. batch_matmul_op.cc
  13. batch_matmul_op.cu
  14. batch_matmul_op.h
  15. boolean_mask_ops.cc
  16. boolean_unmask_ops.cc
  17. cast_op.cc
  18. cast_op.cu
  19. cast_op.h
  20. clip_op.cc
  21. clip_op.cu
  22. clip_op.h
  23. CMakeLists.txt
  24. communicator_op.cc
  25. communicator_op_gpu.cc
  26. concat_split_op.cc
  27. concat_split_op.h
  28. concat_split_op_gpu.cc
  29. conv_gradient_op.cc
  30. conv_op.cc
  31. conv_op.h
  32. conv_op_cache_cudnn.cc
  33. conv_op_cache_cudnn.h
  34. conv_op_cache_cudnn_test.cc
  35. conv_op_cudnn.cc
  36. conv_op_eigen.cc
  37. conv_op_gpu.cc
  38. conv_op_impl.h
  39. conv_op_shared.cc
  40. conv_op_shared.h
  41. conv_op_shared_gpu.cc
  42. conv_pool_op_base.h
  43. conv_transpose_gradient_op.cc
  44. conv_transpose_op.cc
  45. conv_transpose_op.h
  46. conv_transpose_op_cudnn.cc
  47. conv_transpose_op_gpu.cc
  48. conv_transpose_op_impl.h
  49. conv_transpose_op_mobile.cc
  50. conv_transpose_op_mobile.h
  51. conv_transpose_op_mobile_impl.h
  52. conv_transpose_unpool_op_base.h
  53. cosine_embedding_criterion_op.cc
  54. cosine_embedding_criterion_op.cu
  55. cosine_embedding_criterion_op.h
  56. counter_ops.cc
  57. counter_ops.h
  58. counter_ops_gpu.cc
  59. cross_entropy_op.cc
  60. cross_entropy_op.cu
  61. cross_entropy_op.h
  62. dataset_ops.cc
  63. distance_op.cc
  64. distance_op.cu
  65. distance_op.h
  66. dropout_op.cc
  67. dropout_op.cu
  68. dropout_op.h
  69. dropout_op_cudnn.cc
  70. elementwise_add_op.cc
  71. elementwise_div_op.cc
  72. elementwise_linear_op.cc
  73. elementwise_linear_op.cu
  74. elementwise_linear_op.h
  75. elementwise_logical_ops.cc
  76. elementwise_logical_ops.h
  77. elementwise_mul_op.cc
  78. elementwise_op.cc
  79. elementwise_op.cu
  80. elementwise_op.h
  81. elementwise_op_gpu_test.cc
  82. elementwise_op_schema.cc
  83. elementwise_op_test.cc
  84. elementwise_op_test.h
  85. elementwise_sub_op.cc
  86. elu_op.cc
  87. elu_op.h
  88. exp_op.cc
  89. exp_op.cu
  90. extend_tensor_op.cc
  91. feed_blob_op.cc
  92. feed_blob_op.h
  93. filler_op.cc
  94. filler_op.cu
  95. filler_op.h
  96. find_duplicate_elements_op.cc
  97. find_duplicate_elements_op.h
  98. find_op.cc
  99. find_op.cu
  100. find_op.h
  101. free_op.cc
  102. free_op.h
  103. free_op_gpu.cc
  104. fully_connected_op.cc
  105. fully_connected_op.h
  106. fully_connected_op_gpu.cc
  107. fully_connected_op_gpu_test.cc
  108. fully_connected_op_test.cc
  109. given_tensor_fill_op.cc
  110. given_tensor_fill_op.cu
  111. given_tensor_fill_op.h
  112. h_softmax_op.cc
  113. h_softmax_op.h
  114. half_float_ops.cc
  115. half_float_ops.cu
  116. half_float_ops.h
  117. im2col_op.cc
  118. im2col_op.h
  119. im2col_op_gpu.cc
  120. index_ops.cc
  121. instance_norm_gradient_op.cc
  122. instance_norm_op.cc
  123. instance_norm_op.cu
  124. instance_norm_op.h
  125. last_n_window_collector.cc
  126. leaky_relu_op.cc
  127. leaky_relu_op.cu
  128. leaky_relu_op.h
  129. lengths_tile_op.cc
  130. lengths_tile_op.h
  131. load_save_op.cc
  132. load_save_op.h
  133. load_save_op_gpu.cc
  134. local_response_normalization_op.cc
  135. local_response_normalization_op.cu
  136. local_response_normalization_op.h
  137. local_response_normalization_op_cudnn.cc
  138. log_op.cc
  139. log_op.cu
  140. logit_op.cc
  141. loss_op.cc
  142. loss_op.cu
  143. loss_op.h
  144. lp_pool_op.cc
  145. lp_pool_op.cu
  146. lstm_unit_op.cc
  147. lstm_unit_op.h
  148. lstm_unit_op_gpu.cu
  149. margin_ranking_criterion_op.cc
  150. margin_ranking_criterion_op.cu
  151. margin_ranking_criterion_op.h
  152. math_ops.cc
  153. math_ops.cu
  154. math_ops.h
  155. matmul_op.cc
  156. matmul_op.h
  157. matmul_op_gpu.cc
  158. max_pool_with_index.cu
  159. max_pool_with_index.h
  160. mem_query_op.cu
  161. metrics_ops.cc
  162. metrics_ops.h
  163. multi_class_accuracy_op.cc
  164. multi_class_accuracy_op.cu
  165. multi_class_accuracy_op.h
  166. negative_op.cc
  167. negative_op.cu
  168. no_default_engine_op.h
  169. normalize_op.cc
  170. normalize_op.cu
  171. normalize_op.h
  172. one_hot_ops.cc
  173. one_hot_ops.h
  174. operator_fallback_gpu.h
  175. operator_fallback_gpu_test.cc
  176. order_switch_ops.cc
  177. order_switch_ops.cu
  178. order_switch_ops.h
  179. pack_segments.cc
  180. pack_segments.h
  181. pack_segments_op_gpu.cc
  182. pad_op.cc
  183. pad_op.h
  184. pad_op_gpu.cu
  185. partition_ops.cc
  186. partition_ops.h
  187. perplexity_op.cc
  188. perplexity_op.cu
  189. perplexity_op.h
  190. piecewise_linear_transform_op.cc
  191. piecewise_linear_transform_op.h
  192. pool_gradient_op.cc
  193. pool_op.cc
  194. pool_op.cu
  195. pool_op.h
  196. pool_op_cudnn.cc
  197. prefetch_op.h
  198. prelu_op.cc
  199. prelu_op.cu
  200. prelu_op.h
  201. rank_loss_op.cc
  202. rank_loss_op.h
  203. recurrent_network_executor.h
  204. recurrent_network_op.cc
  205. recurrent_network_op.h
  206. recurrent_network_op_gpu.cc
  207. recurrent_op_cudnn.cc
  208. recurrent_op_cudnn.h
  209. reducer_functors.h
  210. reduction_ops.cc
  211. reduction_ops.cu
  212. reduction_ops.h
  213. relu_op.cc
  214. relu_op.cu
  215. relu_op.h
  216. relu_op_cudnn.cc
  217. relu_op_fp16.cu
  218. remove_data_blocks_op.cc
  219. remove_data_blocks_op.h
  220. replace_nan_op.cc
  221. replace_nan_op.h
  222. reshape_op.cc
  223. reshape_op.h
  224. reshape_op_gpu.cc
  225. reshape_op_gpu_test.cc
  226. resize_op.cc
  227. resize_op.h
  228. reverse_packed_segs_op.cc
  229. reverse_packed_segs_op.cu
  230. reverse_packed_segs_op.h
  231. roi_pool_op.cc
  232. roi_pool_op.cu
  233. roi_pool_op.h
  234. rowmul_op.cc
  235. rowmul_op.h
  236. scale_op.cc
  237. scale_op.h
  238. scale_op_gpu.cc
  239. segment_reduction_op.cc
  240. segment_reduction_op.h
  241. segment_reduction_op_gpu.cu
  242. sequence_ops.cc
  243. sequence_ops.h
  244. sigmoid_op.cc
  245. sigmoid_op.cu
  246. softmax_op.cc
  247. softmax_op.h
  248. softmax_op_cudnn.cc
  249. softmax_ops.cu
  250. softmax_shared.cc
  251. softmax_shared.h
  252. softmax_with_loss_op.cc
  253. softmax_with_loss_op.h
  254. softplus_op.cc
  255. softplus_op.cu
  256. softplus_op.h
  257. softsign_op.cc
  258. softsign_op.cu
  259. space_batch_op.cc
  260. space_batch_op.h
  261. space_batch_op_gpu.cu
  262. sparse_to_dense_mask_op.cc
  263. sparse_to_dense_op.cc
  264. sparse_to_dense_op.h
  265. spatial_batch_norm_gradient_op.cc
  266. spatial_batch_norm_op.cc
  267. spatial_batch_norm_op.h
  268. spatial_batch_norm_op_cudnn.cc
  269. spatial_softmax_with_loss_op.cc
  270. spatial_softmax_with_loss_op.h
  271. square_root_divide_op.cc
  272. square_root_divide_op.h
  273. stats_ops.cc
  274. stop_gradient.cc
  275. stop_gradient.h
  276. stop_gradient_gpu.cc
  277. string_ops.cc
  278. string_ops.h
  279. string_ops_test.cc
  280. summarize_op.cc
  281. summarize_op.cu
  282. summarize_op.h
  283. tanh_op.cc
  284. tanh_op.cu
  285. tensor_protos_db_input.cc
  286. tensor_protos_db_input.h
  287. tensor_protos_db_input_gpu.cc
  288. text_file_reader.cc
  289. text_file_reader_utils.cc
  290. text_file_reader_utils.h
  291. text_file_reader_utils_test.cc
  292. tile_op.cc
  293. tile_op.cu
  294. tile_op.h
  295. top_k.cc
  296. top_k.h
  297. top_k_gpu.cc
  298. transpose_op.cc
  299. transpose_op.cu
  300. transpose_op.h
  301. transpose_op_cudnn.cc
  302. tt_linear_op.cc
  303. tt_linear_op.h
  304. utility_ops.cc
  305. utility_ops.cu
  306. utility_ops.h
  307. utility_ops_gpu.cc
  308. utility_ops_gpu_test.cc
  309. utility_ops_test.cc
  310. workspace_ops.cc
  311. zero_gradient_op.cc
  312. zero_gradient_op.h