| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 483 // ConvolutionDescriptor convolution_dimensions;
508 class ConvolutionDescriptor {
513 ConvolutionDescriptor();
514 explicit ConvolutionDescriptor(int ndims);
515 ~ConvolutionDescriptor();
521 ConvolutionDescriptor& set_zero_padding_height(int64 value) {
525 ConvolutionDescriptor& set_zero_padding_width(int64 value) {
529 ConvolutionDescriptor& set_zero_padding(DimIndex dim, int64 value) {
533 ConvolutionDescriptor& set_vertical_filter_stride(int64 value) {
537 ConvolutionDescriptor& set_horizontal_filter_stride(int64 value)
[all...] |
| dnn.cc | 453 // -- ConvolutionDescriptor
455 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) {
463 ConvolutionDescriptor::ConvolutionDescriptor()
464 : ConvolutionDescriptor(/*ndims=*/2) {}
466 ConvolutionDescriptor::~ConvolutionDescriptor() {}
468 string ConvolutionDescriptor::ToString() const {
485 string ConvolutionDescriptor::ToShortString() const
[all...] |
| stream.h | 69 class ConvolutionDescriptor;
280 const dnn::ConvolutionDescriptor &convolution_descriptor,
290 const dnn::ConvolutionDescriptor &convolution_descriptor,
300 const dnn::ConvolutionDescriptor &convolution_descriptor,
309 const dnn::ConvolutionDescriptor &convolution_descriptor,
320 const dnn::ConvolutionDescriptor &convolution_descriptor,
331 const dnn::ConvolutionDescriptor &convolution_descriptor,
342 const dnn::ConvolutionDescriptor &convolution_descriptor,
356 const dnn::ConvolutionDescriptor &convolution_descriptor,
370 const dnn::ConvolutionDescriptor &convolution_descriptor
[all...] |
| stream.cc | 46 string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) {
445 const dnn::ConvolutionDescriptor &convolution_descriptor,
483 const dnn::ConvolutionDescriptor &convolution_descriptor,
521 const dnn::ConvolutionDescriptor &convolution_descriptor,
560 const dnn::ConvolutionDescriptor &convolution_descriptor,
598 const dnn::ConvolutionDescriptor &convolution_descriptor,
640 const dnn::ConvolutionDescriptor &convolution_descriptor,
682 const dnn::ConvolutionDescriptor &convolution_descriptor,
724 const dnn::ConvolutionDescriptor &convolution_descriptor,
740 const dnn::ConvolutionDescriptor &convolution_descriptor
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 269 const dnn::ConvolutionDescriptor& convolution_descriptor,
278 const dnn::ConvolutionDescriptor& convolution_descriptor,
292 const dnn::ConvolutionDescriptor& convolution_descriptor,
307 const dnn::ConvolutionDescriptor& convolution_descriptor,
324 const dnn::ConvolutionDescriptor& convolution_descriptor,
339 const dnn::ConvolutionDescriptor& convolution_descriptor,
352 const dnn::ConvolutionDescriptor& convolution_descriptor,
365 const dnn::ConvolutionDescriptor& convolution_descriptor,
540 const dnn::ConvolutionDescriptor& convolution_descriptor,
588 const dnn::ConvolutionDescriptor& convolution_descriptor
[all...] |
| cuda_dnn.cc | 421 using ConvolutionDescriptor =
452 ConvolutionDescriptor CreateConvolutionDescriptor() {
455 return ConvolutionDescriptor(result);
645 // Turns a ConvolutionDescriptor structure into a cudnn convolution handle
650 const dnn::ConvolutionDescriptor& convolution_descriptor,
709 ConvolutionDescriptor handle_; // Owned.
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 260 const dnn::ConvolutionDescriptor& convolution_descriptor,
269 const dnn::ConvolutionDescriptor& convolution_descriptor,
283 const dnn::ConvolutionDescriptor& convolution_descriptor,
298 const dnn::ConvolutionDescriptor& convolution_descriptor,
315 const dnn::ConvolutionDescriptor& convolution_descriptor,
330 const dnn::ConvolutionDescriptor& convolution_descriptor,
343 const dnn::ConvolutionDescriptor& convolution_descriptor,
356 const dnn::ConvolutionDescriptor& convolution_descriptor,
523 const dnn::ConvolutionDescriptor& convolution_descriptor,
538 const dnn::ConvolutionDescriptor& convolution_descriptor
[all...] |
| rocm_dnn.cc | 61 using dnn::ConvolutionDescriptor;
642 // Turns a ConvolutionDescriptor structure into a miopen convolution handle
647 const ConvolutionDescriptor& convolution_descriptor,
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| cudnn_conv_runner.cc | 35 using se::dnn::ConvolutionDescriptor;
205 ConvolutionDescriptor convolution_descriptor(effective_num_dimensions);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops_3d.cc | 371 se::dnn::ConvolutionDescriptor conv_desc(3);
|
| conv_ops_fused_impl.h | 695 se::dnn::ConvolutionDescriptor conv_desc;
[all...] |
| conv_grad_ops_3d.cc | [all...] |
| conv_grad_filter_ops.cc | 733 se::dnn::ConvolutionDescriptor conv_desc;
[all...] |
| conv_grad_input_ops.cc | [all...] |
| conv_ops.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 611 dnn::ConvolutionDescriptor conv_desc;
[all...] |
