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

  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_dnn.h 543 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
    [all...]
cuda_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_dnn.h 526 bool DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc,
    [all...]
rocm_dnn.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/
dnn.h     [all...]
stream.cc     [all...]

Completed in 957 milliseconds