Searched refs:auxWeightsShape (Results 1 – 2 of 2) sorted by relevance
143 const Shape& auxWeightsShape, const T* recurrentWeightsData, in RNNStep() argument160 aux_input_weights_stride = auxWeightsShape.dimensions[1]; in RNNStep()237 const _Float16* auxWeightsData, const Shape& auxWeightsShape,253 const float* auxWeightsData, const Shape& auxWeightsShape,
59 const T* auxWeightsData, const Shape& auxWeightsShape,