Home
last modified time | relevance | path

Searched refs:FilterDescriptor (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc54 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()
Ddnn.h381 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 …]
Dstream.h69 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 …]
Dstream_executor_pimpl.h373 const dnn::FilterDescriptor &filter_descriptor,
Dstream.cc41 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()
Dstream_executor_pimpl.cc280 const dnn::FilterDescriptor &filter_descriptor, in GetMIOpenConvolveAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h237 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 …]
Drocm_dnn.cc66 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/
Dcuda_dnn.h272 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 …]
Dcuda_dnn.cc432 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/
Dgpu_utils.h223 const se::dnn::FilterDescriptor& filter_desc,
235 const se::dnn::FilterDescriptor& filter_desc,
Dgpu_utils.cc132 const se::dnn::FilterDescriptor& filter_desc, in LogConvAutotuneResults()
175 const se::dnn::FilterDescriptor& filter_desc, in LogFusedConvForwardAutotuneResults()
Dconv_grad_input_ops.cc261 se::dnn::FilterDescriptor filter_desc; in operator ()()
Dconv_ops_3d.cc411 se::dnn::FilterDescriptor filter_desc(3); in launch()
Dconv_ops_fused_impl.h578 se::dnn::FilterDescriptor filter_desc;
Dconv_grad_ops_3d.cc1312 se::dnn::FilterDescriptor filter_desc(3); in Compute()
1809 se::dnn::FilterDescriptor filter_desc(3); in Compute()
Dconv_grad_filter_ops.cc866 se::dnn::FilterDescriptor filter_desc; in operator ()()
Dconv_ops.cc891 se::dnn::FilterDescriptor filter_desc; in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h71 se::dnn::FilterDescriptor filter_descriptor;
Dgpu_conv_runner.cc39 using se::dnn::FilterDescriptor;
387 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig()
388 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()
/external/icu/tools/srcgen/currysrc/libs/
Dorg.eclipse.core.resources_3.13.0.v20180512-1138.jarMETA-INF/MANIFEST.MF META-INF/ECLIPSE_.SF META-INF/ECLIPSE_ ...