Searched refs:miopen_type (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2774 int miopen_type, // Actually miopenDataType_t. in MaybeTransformLayout() argument
2788 *output_descriptor, static_cast<miopenDataType_t>(miopen_type)}; in MaybeTransformLayout()
2791 static_cast<miopenDataType_t>(miopen_type)}; in MaybeTransformLayout()
3228 Stream* stream, int miopen_type, // Actually miopenDataType_t. in DoConvolveBackwardBiasImpl() argument
3236 static_cast<miopenDataType_t>(miopen_type)}; in DoConvolveBackwardBiasImpl()
3238 static_cast<miopenDataType_t>(miopen_type)}; in DoConvolveBackwardBiasImpl()
4022 int miopen_type, // Actually miopenDataType_t. in DoFusedConvolutionBiasActivationImpl() argument
4035 conv_input_descriptor, static_cast<miopenDataType_t>(miopen_type)}; in DoFusedConvolutionBiasActivationImpl()
4038 static_cast<miopenDataType_t>(miopen_type)}; in DoFusedConvolutionBiasActivationImpl()
4041 static_cast<miopenDataType_t>(miopen_type)}; in DoFusedConvolutionBiasActivationImpl()
[all …]
|
| D | rocm_dnn.h | 656 int miopen_type, // Actually miopenDataType_t.
707 int miopen_type, // Actually miopenDataType_t.
721 int miopen_type, // Actually miopenDataType_t.
732 int miopen_type, // Actually miopenDataType_t.
745 int miopen_type, // Actually miopenDataType_t.
|