Searched refs:QuantizedActivationMode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 80 std::string QuantizedActivationModeString(QuantizedActivationMode mode) { in QuantizedActivationModeString()
82 case dnn::QuantizedActivationMode::k8Bit: in QuantizedActivationModeString()
84 case dnn::QuantizedActivationMode::k16Bit: in QuantizedActivationModeString()
86 case dnn::QuantizedActivationMode::k32Bit: in QuantizedActivationModeString()
272 quantized_activation_mode_(QuantizedActivationMode::k8Bit) { in BatchDescriptor()
342 if (quantized_activation_mode() == QuantizedActivationMode::k16Bit) { in ToShortString()
405 CHECK(quantized_activation_mode_ == QuantizedActivationMode::k8Bit); in ToProto()
|
| D | dnn.h | 108 enum class QuantizedActivationMode { enum
189 std::string QuantizedActivationModeString(QuantizedActivationMode mode);
259 QuantizedActivationMode quantized_activation_mode() const { in quantized_activation_mode()
304 QuantizedActivationMode quantized_activation_mode) { in set_quantized_activation_mode()
350 QuantizedActivationMode quantized_activation_mode_;
2063 QuantizedActivationMode mode, void* host_dst, int64 size) = 0;
2084 QuantizedActivationMode mode,
|
| D | stream.h | 700 dnn::QuantizedActivationMode mode,
717 dnn::QuantizedActivationMode mode,
2156 static constexpr dnn::QuantizedActivationMode kModeId =
2157 dnn::QuantizedActivationMode::k8Bit;
2162 static constexpr dnn::QuantizedActivationMode kModeId =
2163 dnn::QuantizedActivationMode::k16Bit;
2168 static constexpr dnn::QuantizedActivationMode kModeId =
2169 dnn::QuantizedActivationMode::k32Bit;
|
| D | stream.cc | 69 std::string ToVlogString(dnn::QuantizedActivationMode mode) { in ToVlogString()
1262 dnn::QuantizedActivationMode mode, void *host_dst, uint64 size) { in ThenMemcpyD2HQuantized()
1276 const void *host_src, uint64 size, dnn::QuantizedActivationMode mode, in ThenMemcpyH2DQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 546 dnn::QuantizedActivationMode mode, void* host_dst,
551 dnn::QuantizedActivationMode mode,
|
| D | cuda_dnn.cc | 4609 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
4616 dnn::QuantizedActivationMode mode, in DoMemcpyH2DQuantized()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 562 dnn::QuantizedActivationMode mode, void* host_dst,
567 dnn::QuantizedActivationMode mode,
|
| D | rocm_dnn.cc | 4570 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
4577 dnn::QuantizedActivationMode mode, in DoMemcpyH2DQuantized()
|