Home
last modified time | relevance | path

Searched refs:filter_desc (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgpu_utils.cc132 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()
Dconv_grad_input_ops.cc261 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 ()()
Dconv_ops_3d.cc411 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()
Dconv_grad_ops_3d.cc1312 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 …]
Dgpu_utils.h223 const se::dnn::FilterDescriptor& filter_desc,
235 const se::dnn::FilterDescriptor& filter_desc,
Dconv_grad_filter_ops.cc866 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 ()()
Dconv_ops.cc891 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 ()()
Dconv_ops_fused_impl.h578 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/
Dcuda_dnn.cc1286 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()