Home
last modified time | relevance | path

Searched refs:QuantizedActivationMode (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc51 string QuantizedActivationModeString(QuantizedActivationMode mode) { in QuantizedActivationModeString()
53 case dnn::QuantizedActivationMode::k8Bit: in QuantizedActivationModeString()
55 case dnn::QuantizedActivationMode::k16Bit: in QuantizedActivationModeString()
57 case dnn::QuantizedActivationMode::k32Bit: in QuantizedActivationModeString()
241 quantized_activation_mode_(QuantizedActivationMode::k8Bit) { in BatchDescriptor()
311 if (quantized_activation_mode() == QuantizedActivationMode::k16Bit) { in ToShortString()
374 CHECK(quantized_activation_mode_ == QuantizedActivationMode::k8Bit); in ToProto()
Ddnn.h92 enum class QuantizedActivationMode { enum
197 string QuantizedActivationModeString(QuantizedActivationMode mode);
267 QuantizedActivationMode quantized_activation_mode() const {
312 QuantizedActivationMode quantized_activation_mode) {
358 QuantizedActivationMode quantized_activation_mode_;
1997 QuantizedActivationMode mode, void* host_dst, int64 size) = 0;
2018 QuantizedActivationMode mode,
Dstream.h677 dnn::QuantizedActivationMode mode,
694 dnn::QuantizedActivationMode mode,
2050 static constexpr dnn::QuantizedActivationMode kModeId =
2051 dnn::QuantizedActivationMode::k8Bit;
2056 static constexpr dnn::QuantizedActivationMode kModeId =
2057 dnn::QuantizedActivationMode::k16Bit;
2062 static constexpr dnn::QuantizedActivationMode kModeId =
2063 dnn::QuantizedActivationMode::k32Bit;
Dstream.cc70 string ToVlogString(dnn::QuantizedActivationMode mode) { in ToVlogString()
1706 dnn::QuantizedActivationMode mode, void *host_dst, uint64 size) { in ThenMemcpyD2HQuantized()
1722 const void *host_src, uint64 size, dnn::QuantizedActivationMode mode, in ThenMemcpyH2DQuantized()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h527 dnn::QuantizedActivationMode mode, void* host_dst,
532 dnn::QuantizedActivationMode mode,
Dcuda_dnn.cc4076 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
4083 dnn::QuantizedActivationMode mode, in DoMemcpyH2DQuantized()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h510 dnn::QuantizedActivationMode mode, void* host_dst,
515 dnn::QuantizedActivationMode mode,
Drocm_dnn.cc3973 dnn::QuantizedActivationMode mode, void* host_dst, int64 size) { in DoMemcpyD2HQuantized()
3980 dnn::QuantizedActivationMode mode, in DoMemcpyH2DQuantized()