HomeSort by relevance Sort by last modified time
    Searched refs:QuantizedActivationMode (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 51 string QuantizedActivationModeString(QuantizedActivationMode mode) {
53 case dnn::QuantizedActivationMode::k8Bit:
55 case dnn::QuantizedActivationMode::k16Bit:
57 case dnn::QuantizedActivationMode::k32Bit:
241 quantized_activation_mode_(QuantizedActivationMode::k8Bit) {
311 if (quantized_activation_mode() == QuantizedActivationMode::k16Bit) {
374 CHECK(quantized_activation_mode_ == QuantizedActivationMode::k8Bit);
dnn.h 92 enum class QuantizedActivationMode {
197 string QuantizedActivationModeString(QuantizedActivationMode mode);
254 // quantized_activation_mode_ == QuantizedActivationMode::k8Bit
267 QuantizedActivationMode quantized_activation_mode() const {
312 QuantizedActivationMode quantized_activation_mode) {
358 QuantizedActivationMode quantized_activation_mode_;
    [all...]
stream.h 77 // Convert a type to the corresponding QuantizedActivationMode.
677 dnn::QuantizedActivationMode mode,
    [all...]
stream.cc 70 string ToVlogString(dnn::QuantizedActivationMode mode) {
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 527 dnn::QuantizedActivationMode mode, void* host_dst,
532 dnn::QuantizedActivationMode mode,
    [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.h 510 dnn::QuantizedActivationMode mode, void* host_dst,
515 dnn::QuantizedActivationMode mode,
    [all...]
rocm_dnn.cc     [all...]

Completed in 114 milliseconds