Searched refs:RnnModelDims (Results 1 – 2 of 2) sorted by relevance
1401 struct RnnModelDims { struct1411 port::StatusOr<RnnModelDims> ExtractAndCheckRnnForward( in ExtractAndCheckRnnForward()1426 RnnModelDims model_dims; in ExtractAndCheckRnnForward()1524 RnnModelDims model_dims, in DoRnnForwardImpl()1684 RnnModelDims model_dims, in DoRnnBackwardImpl()
1881 struct RnnModelDims { struct1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward()2018 RnnModelDims model_dims; in DoRnnForwardImpl()2140 RnnModelDims model_dims; in DoRnnBackwardImpl()