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

  /external/tensorflow/tensorflow/stream_executor/
dnn.h 100 struct ToDataType;
102 struct ToDataType<float> {
106 struct ToDataType<double> {
110 struct ToDataType<Eigen::half> {
114 struct ToDataType<int8> {
118 struct ToDataType<int32> {
    [all...]
stream.h     [all...]
  /external/tensorflow/tensorflow/core/kernels/
cudnn_rnn_ops.cc 132 using se::dnn::ToDataType;
657 se::dnn::DataType data_type = ToDataType<T>::value;
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.cc     [all...]

Completed in 1454 milliseconds