Searched defs:state_desc (Results 1 – 2 of 2) sorted by relevance
653 std::unique_ptr<RnnStateTensorDescriptor>* state_desc, in CreateForwardAndBackwardIODescriptors()742 std::unique_ptr<RnnStateTensorDescriptor> state_desc; in DoForward() local804 std::unique_ptr<RnnStateTensorDescriptor> state_desc; in DoBackward() local
2314 std::unique_ptr<MIOpenRnnStateTensorDescriptor> state_desc( in createRnnStateTensorDescriptor() local