Home
last modified time | relevance | path

Searched refs:RnnModelDims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1401 struct RnnModelDims { struct
1411 port::StatusOr<RnnModelDims> ExtractAndCheckRnnForward( in ExtractAndCheckRnnForward()
1426 RnnModelDims model_dims; in ExtractAndCheckRnnForward()
1524 RnnModelDims model_dims, in DoRnnForwardImpl()
1684 RnnModelDims model_dims, in DoRnnBackwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1881 struct RnnModelDims { struct
1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward()
2018 RnnModelDims model_dims; in DoRnnForwardImpl()
2140 RnnModelDims model_dims; in DoRnnBackwardImpl()