| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.cc | 114 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneFusedConv() argument
140 output_desc, conv_desc, /*use_fallback=*/false, activation_mode, in AutotuneFusedConv()
161 conv_desc, conv_scale, side_input_scale, activation_mode, in AutotuneFusedConv()
191 output_desc, conv_desc, /*use_fallback=*/true, activation_mode, in AutotuneFusedConv()
202 conv_desc, conv_scale, side_input_scale, activation_mode, in AutotuneFusedConv()
229 const se::dnn::ConvolutionDescriptor& conv_desc,
245 const se::dnn::ConvolutionDescriptor& conv_desc,
262 const se::dnn::ConvolutionDescriptor& conv_desc,
279 const se::dnn::ConvolutionDescriptor& conv_desc, in AutotuneUnfusedConv() argument
321 conv_desc, /*use_fallback=*/false, &rz_allocator, &runners)); in AutotuneUnfusedConv()
[all …]
|
| D | conv_ops_gpu.h | 109 const se::dnn::ConvolutionDescriptor& conv_desc,
124 const se::dnn::ConvolutionDescriptor& conv_desc,
168 const se::dnn::ConvolutionDescriptor& conv_desc, in LaunchAutotunedConv() argument
176 conv_desc}; in LaunchAutotunedConv()
197 conv_desc, scratch_allocator, autotune_entry.GetAlgorithmConfig(), in LaunchAutotunedConv()
|
| D | gpu_utils.cc | 145 const se::dnn::ConvolutionDescriptor& conv_desc, in LogConvAutotuneResults() argument
155 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogConvAutotuneResults()
188 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults() argument
198 *instr.mutable_conv_desc() = conv_desc.ToProto(); in LogFusedConvForwardAutotuneResults()
|
| D | conv_grad_input_ops.cc | 263 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
264 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
375 conv_desc.group_count() // group_count in operator ()()
381 filter_desc, filter_ptr, conv_desc, output_desc, out_backprop_ptr, in operator ()()
391 conv_desc, output_desc, out_backprop_ptr); in operator ()()
|
| D | conv_ops_3d.cc | 426 se::dnn::ConvolutionDescriptor conv_desc(3); in launch() local
427 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
500 conv_desc.group_count()}; in launch()
509 filter_ptr, conv_desc, output_desc, output_ptr, ConvolveScratchSize); in launch()
516 stream, input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
|
| D | conv_ops_fused_impl.h | 510 se::dnn::ConvolutionDescriptor conv_desc;
511 conv_desc.set_vertical_dilation_rate(dimensions.dilation_rows)
604 conv_desc.group_count(),
614 output_desc, conv_desc, dnn_activation_mode, kConvScale,
635 conv_desc,
664 conv_desc, // conv
|
| D | conv_grad_ops_3d.cc | 1421 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1422 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1515 conv_desc.group_count()}; in Compute()
1524 input_desc, in_backprop_ptr, filter_desc, filter_ptr, conv_desc, in Compute()
1534 in_backprop_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in Compute()
1820 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1821 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1914 conv_desc.group_count()}; in Compute()
1923 input_desc, input_ptr, filter_desc, filter_backprop_ptr, conv_desc, in Compute()
1933 input_ptr, filter_desc, filter_backprop_ptr, conv_desc, output_desc, in Compute()
|
| D | conv_grad_filter_ops.cc | 882 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
883 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
984 conv_desc.group_count() // group_count in operator ()()
990 filter_desc, filter_backprop_ptr, conv_desc, output_desc, in operator ()()
999 filter_desc, filter_backprop_ptr, conv_desc, output_desc, in operator ()()
|
| D | gpu_utils.h | 264 const se::dnn::ConvolutionDescriptor& conv_desc,
276 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale,
|
| D | conv_ops.cc | 1048 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
1049 conv_desc.set_vertical_dilation_rate(row_dilation) in operator ()()
1132 conv_desc.group_count()}; in operator ()()
1137 filter_ptr, conv_desc, output_desc, output_ptr, ConvolveScratchSize); in operator ()()
1144 stream, input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 107 params.config->conv_desc}; in RunGpuConvUnfused()
168 params.config->conv_desc, in RunGpuConvForwardActivation()
421 config.conv_desc = ConvolutionDescriptor(effective_num_dimensions); in GetGpuConvConfig()
422 config.conv_desc.set_group_count(desc.feature_group_count); in GetGpuConvConfig()
423 config.conv_desc.set_convolution_not_crosscorr(dims_reversed); in GetGpuConvConfig()
425 config.conv_desc in GetGpuConvConfig()
455 config.conv_desc.set_zero_padding(static_cast<DimIndex>(dim), 0) in GetGpuConvConfig()
|
| D | gpu_conv_runner.h | 64 se::dnn::ConvolutionDescriptor conv_desc; member
|
| D | gpu_conv_algorithm_picker.cc | 144 config.output_descriptor, config.conv_desc, use_fallback, in GetAlgorithms()
169 /* output_data = */ DeviceMemoryBase(nullptr), config.conv_desc, in GetAlgorithms()
208 params.config->conv_desc, /* use_fallback = */ false, scratch_allocator, in GetMIOpenAlgorithms()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 18 stream_executor.dnn.ConvolutionDescriptorProto conv_desc = 5; field
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3569 auto conv_desc = in GetCudnnOperationGraph() local
3579 RETURN_MSG_IF_CUDNN_ERROR(conv_desc); in GetCudnnOperationGraph()
3589 .setcDesc(conv_desc) in GetCudnnOperationGraph()
3606 << "\nConv: " << conv_desc.describe() << "\nOp: " << op.describe() in GetCudnnOperationGraph()
3744 auto conv_desc = in GetCudnnFusedOperationGraph() local
3754 RETURN_MSG_IF_CUDNN_ERROR(conv_desc); in GetCudnnFusedOperationGraph()
3761 .setcDesc(conv_desc) in GetCudnnFusedOperationGraph()
3882 << "\nConv: " << conv_desc.describe() << "\nAdd: " in GetCudnnFusedOperationGraph()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 401 uint64_t GetHashValue(miopenConvolutionDescriptor_t conv_desc) { in GetHashValue() argument
404 wrap::miopenGetConvolutionNdDescriptor(conv_desc, 0, &nd, nullptr, nullptr, in GetHashValue()
412 conv_desc, nd, &nd, pad.data(), stride.data(), dilation.data(), &c_mode); in GetHashValue()
|