| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 54 const dnn::FilterDescriptor& /*filter_descriptor*/, in GetMIOpenConvolveAlgorithms() argument
414 FilterDescriptor::FilterDescriptor(int ndims) { in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
419 FilterDescriptor::FilterDescriptor() : FilterDescriptor(/*ndims=*/2) {} in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
421 FilterDescriptor::~FilterDescriptor() {} in ~FilterDescriptor()
423 void FilterDescriptor::CloneFrom(const FilterDescriptor& other) { in CloneFrom()
427 std::string FilterDescriptor::ToString() const { in ToString()
441 std::string FilterDescriptor::ToShortString() const { in ToShortString()
470 int64 FilterDescriptor::ComputeWeightCount() const { in ComputeWeightCount()
478 TensorDescriptorProto FilterDescriptor::ToProto(DataType data_type) const { in ToProto()
|
| D | dnn.h | 381 class FilterDescriptor {
386 FilterDescriptor();
387 explicit FilterDescriptor(int ndims);
388 ~FilterDescriptor();
391 FilterDescriptor& set_output_feature_map_count(int64 value) { in set_output_feature_map_count()
395 FilterDescriptor& set_input_feature_map_count(int64 value) { in set_input_feature_map_count()
399 FilterDescriptor& set_input_filter_height(int64 value) { in set_input_filter_height()
403 FilterDescriptor& set_input_filter_width(int64 value) { in set_input_filter_width()
407 FilterDescriptor& set_layout(FilterLayout layout) { in set_layout()
411 FilterDescriptor& set_spatial_dim(DimIndex dim, int64 value) { in set_spatial_dim()
[all …]
|
| D | stream.h | 69 class FilterDescriptor; variable
299 const dnn::FilterDescriptor &filter_descriptor,
308 const dnn::FilterDescriptor &filter_descriptor,
318 const dnn::FilterDescriptor &filter_descriptor,
329 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm()
357 const dnn::FilterDescriptor &filter_descriptor,
371 const dnn::FilterDescriptor &filter_descriptor,
385 const dnn::FilterDescriptor &filter_descriptor,
400 const dnn::FilterDescriptor &filter_descriptor,
414 const dnn::FilterDescriptor &filter_descriptor,
[all …]
|
| D | stream_executor_pimpl.h | 373 const dnn::FilterDescriptor &filter_descriptor,
|
| D | stream.cc | 41 std::string ToVlogString(const dnn::FilterDescriptor &descriptor) { in ToVlogString()
449 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
480 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
511 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
543 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
574 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
605 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
624 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
651 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
678 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier, in ThenSeparableConvolve()
|
| D | stream_executor_pimpl.cc | 280 const dnn::FilterDescriptor &filter_descriptor, in GetMIOpenConvolveAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 237 const dnn::FilterDescriptor& filter_descriptor,
310 const dnn::FilterDescriptor& filter_descriptor,
321 const dnn::FilterDescriptor& filter_descriptor,
335 const dnn::FilterDescriptor& filter_descriptor,
349 const dnn::FilterDescriptor& filter_descriptor,
365 const dnn::FilterDescriptor& filter_descriptor,
379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
392 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
405 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
574 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| D | rocm_dnn.cc | 66 using dnn::FilterDescriptor;
710 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
2846 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2942 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3100 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
3122 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3331 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3678 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3694 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3710 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 272 const dnn::FilterDescriptor& filter_descriptor,
283 const dnn::FilterDescriptor& filter_descriptor,
297 const dnn::FilterDescriptor& filter_descriptor,
311 const dnn::FilterDescriptor& filter_descriptor,
327 const dnn::FilterDescriptor& filter_descriptor,
341 const dnn::FilterDescriptor& filter_descriptor,
355 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
368 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
381 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
558 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| D | cuda_dnn.cc | 432 using FilterDescriptor = typedef
462 FilterDescriptor CreateFilterDescriptor() { in CreateFilterDescriptor()
465 return FilterDescriptor(result); in CreateFilterDescriptor()
567 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
605 FilterDescriptor handle_; // Owned.
1027 CudnnRnnParamsDescriptor(FilterDescriptor handle, int64 params_size_in_bytes, in CudnnRnnParamsDescriptor()
1048 FilterDescriptor handle_;
1286 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights()
1287 const FilterDescriptor& region_desc_handle, in CheckAndFetchProjectionWeights()
1376 FilterDescriptor filter_desc = CreateFilterDescriptor(); in Create()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 223 const se::dnn::FilterDescriptor& filter_desc,
235 const se::dnn::FilterDescriptor& filter_desc,
|
| D | gpu_utils.cc | 132 const se::dnn::FilterDescriptor& filter_desc, in LogConvAutotuneResults()
175 const se::dnn::FilterDescriptor& filter_desc, in LogFusedConvForwardAutotuneResults()
|
| D | conv_grad_input_ops.cc | 261 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| D | conv_ops_3d.cc | 411 se::dnn::FilterDescriptor filter_desc(3); in launch()
|
| D | conv_ops_fused_impl.h | 578 se::dnn::FilterDescriptor filter_desc;
|
| D | conv_grad_ops_3d.cc | 1312 se::dnn::FilterDescriptor filter_desc(3); in Compute()
1809 se::dnn::FilterDescriptor filter_desc(3); in Compute()
|
| D | conv_grad_filter_ops.cc | 866 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| D | conv_ops.cc | 891 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 71 se::dnn::FilterDescriptor filter_descriptor;
|
| D | gpu_conv_runner.cc | 39 using se::dnn::FilterDescriptor;
387 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig()
388 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()
|