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

  /external/tensorflow/tensorflow/core/util/
use_cudnn.h 23 // FP16ConvMode: The mode to set the internal compute type for cudnn convolution
27 enum class FP16ConvMode {
35 FP16ConvMode CudnnConvComputeMode();
use_cudnn.cc 42 FP16ConvMode CudnnConvComputeMode() {
50 return FP16ConvMode::kAccurate;
52 return FP16ConvMode::kFast;
54 LOG(ERROR) << "FP16ConvMode only supports two modes, ACCURATE and FAST. "
58 return FP16ConvMode::kAccurate;

Completed in 53 milliseconds