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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
convolution_thunk.h 102 bool tensor_ops_enabled_; member in class:xla::gpu::ConvolutionThunk
  /external/tensorflow/tensorflow/stream_executor/
dnn.h 705 AlgorithmDesc() : algo_(kDefaultAlgorithm), tensor_ops_enabled_(false) {}
707 : algo_(a), tensor_ops_enabled_(use_tensor_ops) {}
709 bool tensor_ops_enabled() const { return tensor_ops_enabled_; }
713 this->tensor_ops_enabled_ == other.tensor_ops_enabled_;
719 bool tensor_ops_enabled_; member in class:perftools::gputools::dnn::AlgorithmDesc
    [all...]

Completed in 481 milliseconds