| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.h | 90 StatusOr<se::dnn::ConvolutionKind> GetDNNConvKindFromCudnnConvKind(
|
| D | stream_executor_util.cc | 322 StatusOr<se::dnn::ConvolutionKind> GetDNNConvKindFromCudnnConvKind( in GetDNNConvKindFromCudnnConvKind()
|
| D | gpu_conv_algorithm_picker.cc | 133 TF_ASSIGN_OR_RETURN(se::dnn::ConvolutionKind kind, in GetMIOpenAlgorithms()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 14 stream_executor.dnn.ConvolutionKind kind = 1;
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 515 dnn::ConvolutionKind kind, miopenConvAlgoPerf_t algorithm) { in GetProfileResultFromConvAlgoPerf()
518 case dnn::ConvolutionKind::FORWARD: in GetProfileResultFromConvAlgoPerf()
522 case dnn::ConvolutionKind::BACKWARD_DATA: in GetProfileResultFromConvAlgoPerf()
526 case dnn::ConvolutionKind::BACKWARD_FILTER: in GetProfileResultFromConvAlgoPerf()
2844 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream, in DoPrepareForConvolution()
2939 dnn::ConvolutionKind kind, dnn::DataType element_type, in DoConvolve()
2982 case dnn::ConvolutionKind::FORWARD: { in DoConvolve()
3001 case dnn::ConvolutionKind::BACKWARD_DATA: { in DoConvolve()
3027 case dnn::ConvolutionKind::BACKWARD_FILTER: { in DoConvolve()
3098 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream, in GetMIOpenConvolveAlgorithms()
[all …]
|
| D | rocm_dnn.h | 235 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
307 dnn::ConvolutionKind kind, dnn::DataType element_type,
834 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
865 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
876 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 126 void LogConvAutotuneResults(se::dnn::ConvolutionKind kind, in LogConvAutotuneResults()
183 instr.set_kind(se::dnn::ConvolutionKind::FORWARD_BIAS_ACTIVATION); in LogFusedConvForwardAutotuneResults()
|
| D | gpu_utils.h | 217 void LogConvAutotuneResults(se::dnn::ConvolutionKind kind,
|
| D | conv_grad_input_ops.cc | 447 se::dnn::ConvolutionKind::BACKWARD_DATA, in operator ()()
493 se::dnn::ConvolutionKind::BACKWARD_DATA, se::dnn::ToDataType<T>::value, in operator ()()
|
| D | conv_ops_3d.cc | 569 se::dnn::ConvolutionKind::FORWARD, in launch()
616 LogConvAutotuneResults(se::dnn::ConvolutionKind::FORWARD, in launch()
|
| D | conv_grad_ops_3d.cc | 1484 se::dnn::ConvolutionKind::BACKWARD_DATA, in Compute()
1511 LogConvAutotuneResults(se::dnn::ConvolutionKind::BACKWARD_DATA, in Compute()
1960 se::dnn::ConvolutionKind::BACKWARD_FILTER, in Compute()
1988 LogConvAutotuneResults(se::dnn::ConvolutionKind::BACKWARD_FILTER, in Compute()
|
| D | conv_grad_filter_ops.cc | 1045 se::dnn::ConvolutionKind::BACKWARD_FILTER, in operator ()()
1090 LogConvAutotuneResults(se::dnn::ConvolutionKind::BACKWARD_FILTER, in operator ()()
|
| D | conv_ops.cc | 1057 se::dnn::ConvolutionKind::FORWARD, se::dnn::ToDataType<T>::value, in operator ()()
1102 LogConvAutotuneResults(se::dnn::ConvolutionKind::FORWARD, in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1242 ConvolutionKind kind, Stream* stream, in PrepareForConvolution()
1295 ConvolutionKind kind, DataType element_type, DataType output_type,
1316 DoConvolve(ConvolutionKind::FORWARD, ToDataType<ElementType>::value, in DoConvolve()
1331 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
1420 ConvolutionKind::BACKWARD_DATA, ToDataType<ElementType>::value, in DoConvolveBackwardData()
1467 ConvolutionKind::BACKWARD_FILTER, ToDataType<ElementType>::value, in DoConvolveBackwardFilter()
2680 ConvolutionKind kind, DataType element_type, Stream* stream, in DoPrepareForConvolution()
|
| D | dnn.proto | 73 enum ConvolutionKind { enum
|
| D | stream.h | 340 dnn::ConvolutionKind::FORWARD, this, input_descriptor, input_data, in ConvolveWithAlgorithm()
345 dnn::ConvolutionKind::FORWARD, dnn::ToDataType<InputType>::value, in ConvolveWithAlgorithm()
451 dnn::ConvolutionKind::BACKWARD_DATA, this, input_descriptor, in ConvolveBackwardDataWithAlgorithm()
457 dnn::ConvolutionKind::BACKWARD_DATA, in ConvolveBackwardDataWithAlgorithm()
483 dnn::ConvolutionKind::BACKWARD_FILTER, this, input_descriptor, in ConvolveBackwardFilterWithAlgorithm()
489 dnn::ConvolutionKind::BACKWARD_FILTER, in ConvolveBackwardFilterWithAlgorithm()
|
| D | dnn.cc | 51 dnn::ConvolutionKind /*kind*/, dnn::DataType /*element_type*/, in GetMIOpenConvolveAlgorithms() argument
|
| D | stream_executor_pimpl.h | 371 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream *stream,
|
| D | stream_executor_pimpl.cc | 278 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream *stream, in GetMIOpenConvolveAlgorithms()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2984 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream, in DoPrepareForConvolution()
3006 case dnn::ConvolutionKind::FORWARD: { in DoPrepareForConvolution()
3014 case dnn::ConvolutionKind::BACKWARD_DATA: { in DoPrepareForConvolution()
3022 case dnn::ConvolutionKind::BACKWARD_FILTER: { in DoPrepareForConvolution()
3039 dnn::ConvolutionKind kind, dnn::DataType element_type, in DoConvolve()
3108 case dnn::ConvolutionKind::FORWARD: { in DoConvolve()
3121 case dnn::ConvolutionKind::BACKWARD_DATA: { in DoConvolve()
3139 case dnn::ConvolutionKind::BACKWARD_FILTER: { in DoConvolve()
|
| D | cuda_dnn.h | 269 dnn::ConvolutionKind kind, dnn::DataType element_type,
695 dnn::ConvolutionKind kind, dnn::DataType element_type, Stream* stream,
|