| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.cc | 121 const dnn::FilterDescriptor& /*filter_descriptor*/, in GetConvolveRunners() argument
136 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc()
148 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners()
164 const dnn::FilterDescriptor& filter_descriptor, in FusedConvolveRunnerFromDesc()
177 const dnn::FilterDescriptor& /*filter_descriptor*/, in GetMIOpenConvolveAlgorithms() argument
623 FilterDescriptor::FilterDescriptor(int ndims) { in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
628 FilterDescriptor::FilterDescriptor() : FilterDescriptor(/*ndims=*/2) {} in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
630 FilterDescriptor::~FilterDescriptor() {} in ~FilterDescriptor()
632 void FilterDescriptor::CloneFrom(const FilterDescriptor& other) { in CloneFrom()
636 std::string FilterDescriptor::ToString() const { in ToString()
[all …]
|
| H A D | dnn.h | 401 class FilterDescriptor {
406 FilterDescriptor();
407 explicit FilterDescriptor(int ndims);
408 ~FilterDescriptor();
411 FilterDescriptor& set_output_feature_map_count(int64_t value) { in set_output_feature_map_count()
415 FilterDescriptor& set_input_feature_map_count(int64_t value) { in set_input_feature_map_count()
419 FilterDescriptor& set_input_filter_height(int64_t value) { in set_input_filter_height()
423 FilterDescriptor& set_input_filter_width(int64_t value) { in set_input_filter_width()
427 FilterDescriptor& set_layout(FilterLayout layout) { in set_layout()
431 FilterDescriptor& set_spatial_dim(DimIndex dim, int64_t value) { in set_spatial_dim()
[all …]
|
| H A D | lazy_op_runner.h | 124 const FilterDescriptor& filter_descriptor;
148 const FilterDescriptor& filter_descriptor;
|
| H A D | stream.h | 68 class FilterDescriptor; variable
309 const dnn::FilterDescriptor &filter_descriptor,
318 const dnn::FilterDescriptor &filter_descriptor,
328 const dnn::FilterDescriptor &filter_descriptor,
339 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm()
370 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
397 const dnn::FilterDescriptor &filter_descriptor, in ConvolveRunnerFromDesc()
416 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveRunnerFromDesc()
435 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier,
|
| H A D | stream_executor_pimpl.h | 376 const dnn::FilterDescriptor& filter_descriptor,
390 const dnn::FilterDescriptor& filter_descriptor,
402 const dnn::FilterDescriptor& filter_descriptor,
|
| H A D | stream_executor_pimpl.cc | 297 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners()
320 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners()
340 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
|
| H A D | stream.cc | 43 std::string ToVlogString(const dnn::FilterDescriptor &descriptor) { in ToVlogString()
440 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
460 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
487 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
514 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier, in ThenSeparableConvolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 244 const dnn::FilterDescriptor& filter_descriptor,
257 const dnn::FilterDescriptor& filter_descriptor,
264 const dnn::FilterDescriptor& filter_descriptor,
343 const dnn::FilterDescriptor& filter_descriptor,
356 const dnn::FilterDescriptor& filter_descriptor,
370 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
383 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
396 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
511 const dnn::FilterDescriptor& filter_descriptor,
525 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| H A D | rocm_dnn.cc | 70 using dnn::FilterDescriptor;
714 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
2906 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2958 FilterDescriptor filter_descriptor, in RocmConvRunner()
3115 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3150 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners()
3190 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc()
3215 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
3237 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3446 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 224 const dnn::FilterDescriptor& filter_descriptor,
237 const dnn::FilterDescriptor& filter_descriptor,
247 const dnn::FilterDescriptor& filter_descriptor,
262 const dnn::FilterDescriptor& filter_descriptor,
339 const dnn::FilterDescriptor& filter_descriptor,
352 const dnn::FilterDescriptor& filter_descriptor,
366 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
392 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
505 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| H A D | cuda_dnn.cc | 529 using FilterDescriptor = typedef
559 FilterDescriptor CreateFilterDescriptor() { in CreateFilterDescriptor()
562 return FilterDescriptor(result); in CreateFilterDescriptor()
670 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
715 FilterDescriptor handle_; // Owned.
1201 CudnnRnnParamsDescriptor(FilterDescriptor handle, in CudnnRnnParamsDescriptor()
1223 FilterDescriptor handle_;
1464 const TensorDescriptor& input_desc, const FilterDescriptor& filter_desc, in CheckAndFetchProjectionWeights()
1465 const FilterDescriptor& region_desc_handle, in CheckAndFetchProjectionWeights()
1554 FilterDescriptor filter_desc = CreateFilterDescriptor(); in Create()
[all …]
|
| /aosp_15_r20/external/golang-protobuf/testing/protocmp/ |
| H A D | util.go | 39 return FilterDescriptor(enum.Descriptor(), opt)
55 return FilterDescriptor(message.ProtoReflect().Descriptor(), opt)
70 return FilterDescriptor(mustFindFieldDescriptor(md, name), opt)
86 return FilterDescriptor(mustFindOneofDescriptor(md, name), opt)
103 func FilterDescriptor(desc protoreflect.Descriptor, opt cmp.Option) cmp.Option { func
544 opt = FilterDescriptor(et.Descriptor(), cmpopts.SortSlices(lessFunc))
557 opt = FilterDescriptor(mt.Descriptor(), cmpopts.SortSlices(lessFunc))
664 opts = append(opts, FilterDescriptor(fd, cmpopts.SortSlices(lessFunc)))
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_gpu.cc | 111 const se::dnn::FilterDescriptor& filter_desc, in AutotuneFusedConv()
226 const se::dnn::FilterDescriptor& filter_desc,
242 const se::dnn::FilterDescriptor& filter_desc,
259 const se::dnn::FilterDescriptor& filter_desc,
277 se::DeviceMemory<T> input_ptr, const se::dnn::FilterDescriptor& filter_desc, in AutotuneUnfusedConv()
448 const se::dnn::FilterDescriptor& filter_desc,
460 const se::dnn::FilterDescriptor& filter_desc,
473 const se::dnn::FilterDescriptor& filter_desc,
|
| H A D | conv_ops_gpu.h | 106 const se::dnn::FilterDescriptor& filter_desc,
122 se::DeviceMemory<T> input_ptr, const se::dnn::FilterDescriptor& filter_desc,
166 const se::dnn::FilterDescriptor& filter_desc, in LaunchAutotunedConv()
|
| H A D | gpu_utils.cc | 143 const se::dnn::FilterDescriptor& filter_desc, in LogConvAutotuneResults()
186 const se::dnn::FilterDescriptor& filter_desc, in LogFusedConvForwardAutotuneResults()
|
| H A D | gpu_utils.h | 262 const se::dnn::FilterDescriptor& filter_desc,
274 const se::dnn::FilterDescriptor& filter_desc,
|
| H A D | conv_grad_input_ops.cc | 257 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
| H A D | Descriptors.cpp | 124 void FilterDescriptor::set(const at::Tensor &t, const at::MemoryFormat memory_format, int64_t pad) { in set()
174 std::ostream& operator<<(std::ostream & out, const FilterDescriptor& d) { in operator <<()
193 void FilterDescriptor::print() { std::cout << *this; } in print()
|
| H A D | Descriptors.h | 176 class TORCH_CUDA_CPP_API FilterDescriptor : public Descriptor<
194 std::ostream& operator<<(std::ostream & out, const FilterDescriptor& d);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | RNN.cpp | 595 const FilterDescriptor& w_desc, in add_projection_weights()
605 FilterDescriptor lin_layer_mat_desc; in add_projection_weights()
703 const FilterDescriptor& w_desc, in get_parameters()
724 FilterDescriptor lin_layer_mat_desc; in get_parameters()
882 FilterDescriptor w_desc; in get_parameters()
909 FilterDescriptor lin_layer_mat_desc; in get_parameters()
957 FilterDescriptor lin_layer_mat_desc; in get_parameters()
1299 FilterDescriptor w_desc; in copy_weights_to_flat_buf_views()
1482 FilterDescriptor w_desc; in _cudnn_rnn()
1809 FilterDescriptor w_desc; in _cudnn_rnn_backward_input()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 331 … const RNNDescriptor& rnn_desc, const TensorDescriptor& x_desc, const FilterDescriptor& w_desc, in get_parameters()
347 FilterDescriptor lin_layer_mat_desc; in get_parameters()
382 FilterDescriptor lin_layer_mat_desc; in get_parameters()
498 FilterDescriptor w_desc; in miopen_rnn()
631 FilterDescriptor w_desc; in miopen_rnn_backward_input()
725 FilterDescriptor w_desc; in miopen_rnn_backward_weight()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_runner.cc | 65 using se::dnn::FilterDescriptor;
407 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig()
408 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()
|
| H A D | gpu_conv_runner.h | 62 se::dnn::FilterDescriptor filter_descriptor;
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | Descriptors.h | 91 class FilterDescriptor
|
| H A D | Descriptors.cpp | 94 void FilterDescriptor::set(const at::Tensor &t, const at::MemoryFormat memory_format, int64_t pad) { in set()
|