/external/tensorflow/tensorflow/core/kernels/ |
D | gpu_utils.cc | 132 const se::dnn::FilterDescriptor& filter_desc, in LogConvAutotuneResults() argument 142 *instr.mutable_filter() = filter_desc.ToProto(element_type); in LogConvAutotuneResults() 175 const se::dnn::FilterDescriptor& filter_desc, in LogFusedConvForwardAutotuneResults() argument 185 *instr.mutable_filter() = filter_desc.ToProto(element_type); in LogFusedConvForwardAutotuneResults()
|
D | conv_grad_input_ops.cc | 261 se::dnn::FilterDescriptor filter_desc; in operator ()() local 262 filter_desc.set_input_filter_height(dims.spatial_dims[0].filter_size) 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 ()() 494 in_backprop_ptr, filter_ptr, out_backprop_ptr, input_desc, filter_desc, in operator ()() 501 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
D | conv_ops_3d.cc | 411 se::dnn::FilterDescriptor filter_desc(3); in launch() local 412 filter_desc.set_spatial_dim(DimIndex::X, filter_cols) in launch() 541 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch() 571 input_ptr, filter_desc, filter_ptr, output_desc, in launch() 595 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch() 618 filter_ptr, output_ptr, input_desc, filter_desc, in launch() 626 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in launch()
|
D | conv_grad_ops_3d.cc | 1312 se::dnn::FilterDescriptor filter_desc(3); in Compute() local 1313 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) 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() 1809 se::dnn::FilterDescriptor filter_desc(3); in Compute() local 1810 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) in Compute() 1940 filter_desc, &filter_backprop_ptr, &scratch_allocator, in Compute() [all …]
|
D | gpu_utils.h | 223 const se::dnn::FilterDescriptor& filter_desc, 235 const se::dnn::FilterDescriptor& filter_desc,
|
D | conv_grad_filter_ops.cc | 866 se::dnn::FilterDescriptor filter_desc; in operator ()() local 867 filter_desc.set_input_filter_height(dims.spatial_dims[0].filter_size) in operator ()() 1017 filter_desc, &filter_backprop_ptr_rz, allocator_used, in operator ()() 1047 filter_desc, filter_backprop_ptr, output_desc, out_backprop_ptr, in operator ()() 1073 filter_desc, &filter_backprop_ptr, &scratch_allocator, in operator ()() 1093 filter_desc, output_desc, conv_desc, in operator ()() 1102 filter_desc, &filter_backprop_ptr, &scratch_allocator, algorithm_config, in operator ()()
|
D | conv_ops.cc | 891 se::dnn::FilterDescriptor filter_desc; in operator ()() local 892 filter_desc.set_input_filter_height(patch_rows) in operator ()() 1028 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()() 1058 stream, input_desc, input_ptr, filter_desc, filter_ptr, output_desc, in operator ()() 1084 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()() 1104 output_tensor, input_desc, filter_desc, output_desc, in operator ()() 1117 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, output_desc, in operator ()()
|
D | conv_ops_fused_impl.h | 578 se::dnn::FilterDescriptor filter_desc; 579 filter_desc.set_input_filter_height(patch_rows) 676 filter_desc, filter_ptr, // filter 692 output_ptr, bias_ptr, side_input_ptr, input_desc, filter_desc,
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 1286 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights() argument 1333 /*wDesc=*/filter_desc.get(), in CheckAndFetchProjectionWeights() 1376 FilterDescriptor filter_desc = CreateFilterDescriptor(); in Create() local 1380 /*filterDesc=*/filter_desc.get(), /*dataType=*/data_type, in Create() 1416 /*wDesc=*/filter_desc.get(), in Create() 1423 /*wDesc=*/filter_desc.get(), in Create() 1444 cudnn, rnn_desc, layer, input_desc, filter_desc, region_desc_handle, in Create() 1448 return CudnnRnnParamsDescriptor(std::move(filter_desc), params_size_in_bytes, in Create()
|