projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 914796d) | patch
Adding a depthwise convolution kernel op (with label 'cudnn_grouped_convolution'...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Mon, 30 Apr 2018 13:59:23 +0000 (06:59 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 30 Apr 2018 14:01:55 +0000 (07:01 -0700)
commita5a51ad3a1200e2e5ef46c140bab717422e41ca2
tree7c1e5ee890228c81ef57d6c6998a04dac79a29e2tree | snapshot
parent914796d5e9bc7b0c619b53c7eb24cfe7d6c7fb9bcommit | diff
Adding a depthwise convolution kernel op (with label 'cudnn_grouped_convolution') which forwards to cuDNN grouped convolutions.

PiperOrigin-RevId: 194780352
tensorflow/core/kernels/BUILD diff | blob | history
tensorflow/core/kernels/conv_grad_filter_ops.cc diff | blob | history
tensorflow/core/kernels/conv_grad_input_ops.cc diff | blob | history
tensorflow/core/kernels/conv_grad_ops.cc diff | blob | history
tensorflow/core/kernels/conv_ops.cc diff | blob | history
tensorflow/core/kernels/depthwise_conv_grad_op.cc diff | blob | history
tensorflow/core/kernels/depthwise_conv_op.cc diff | blob | history
tensorflow/python/kernel_tests/depthwise_conv_op_test.py diff | blob | history
tensorflow/stream_executor/cuda/cuda_dnn.cc diff | blob | history
tensorflow/stream_executor/dnn.cc diff | blob | history
tensorflow/stream_executor/dnn.h diff | blob | history
Domain: Machine Learning / ML Framework;