| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 90 params.config.filter_descriptor, filter_buf, params.config.conv_desc, in RunGpuConvForward()
131 filter_buf, params.config.conv_desc, side_input, in RunGpuConvForwardActivation()
168 params.config.output_descriptor, output_buf, params.config.conv_desc, in RunGpuConvInternalImpl()
180 params.config.output_descriptor, output_buf, params.config.conv_desc, in RunGpuConvInternalImpl()
400 config.conv_desc = ConvolutionDescriptor(effective_num_dimensions); in GetGpuConvConfig()
401 config.conv_desc.set_group_count(desc.feature_group_count); in GetGpuConvConfig()
402 config.conv_desc.set_convolution_not_crosscorr(dims_reversed); in GetGpuConvConfig()
404 config.conv_desc in GetGpuConvConfig()
433 config.conv_desc.set_zero_padding(static_cast<DimIndex>(dim), 0) in GetGpuConvConfig()
|
| D | gpu_conv_runner.h | 73 se::dnn::ConvolutionDescriptor conv_desc; member
|
| D | gpu_conv_algorithm_picker.cc | 146 params.config.conv_desc, scratch_allocator, &algorithms); in GetMIOpenAlgorithms()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 134 const se::dnn::ConvolutionDescriptor& conv_desc, in LogConvAutotuneResults() argument
144 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogConvAutotuneResults()
177 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults() argument
187 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogFusedConvForwardAutotuneResults()
|
| D | conv_grad_input_ops.cc | 267 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
268 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
380 conv_desc.group_count() // group_count in operator ()()
421 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
449 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
474 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
495 output_desc, conv_desc, stream->parent(), results); in operator ()()
501 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_ops_3d.cc | 418 se::dnn::ConvolutionDescriptor conv_desc(3); in launch() local
419 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
493 conv_desc.group_count()}; in launch()
541 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
572 output_ptr, conv_desc, &scratch_allocator, &algorithms), in launch()
595 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
619 output_desc, conv_desc, stream->parent(), results); in launch()
626 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in launch()
|
| D | conv_grad_ops_3d.cc | 1319 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1320 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1413 conv_desc.group_count()}; in Compute()
1455 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1486 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1493 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1514 filter_desc, output_desc, conv_desc, in Compute()
1524 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1816 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1817 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
[all …]
|
| D | conv_grad_filter_ops.cc | 872 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
873 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
975 conv_desc.group_count() // group_count in operator ()()
1016 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1048 conv_desc, &scratch_allocator, &algorithms), in operator ()()
1072 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1093 filter_desc, output_desc, conv_desc, in operator ()()
1101 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | gpu_utils.h | 225 const se::dnn::ConvolutionDescriptor& conv_desc,
237 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale,
|
| D | conv_ops.cc | 897 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
898 conv_desc.set_vertical_dilation_rate(row_dilation) in operator ()()
984 conv_desc.group_count()}; in operator ()()
1028 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
1059 output_ptr, conv_desc, &scratch_allocator, &algorithms), in operator ()()
1084 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
1105 conv_desc, stream->parent(), results); in operator ()()
1117 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in operator ()()
|
| D | conv_ops_fused_impl.h | 589 se::dnn::ConvolutionDescriptor conv_desc;
590 conv_desc.set_vertical_dilation_rate(dimensions.dilation_rows)
663 conv_desc.group_count()},
677 conv_desc, // conv
693 output_desc, conv_desc, 1.0, 0.0, dnn_activation_mode,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 18 stream_executor.dnn.ConvolutionDescriptorProto conv_desc = 5; field
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 397 uint64 GetHashValue(miopenConvolutionDescriptor_t conv_desc) { in GetHashValue() argument
400 wrap::miopenGetConvolutionNdDescriptor(conv_desc, 0, &nd, nullptr, nullptr, in GetHashValue()
408 conv_desc, nd, &nd, pad.data(), stride.data(), dilation.data(), &c_mode); in GetHashValue()
|