Searched refs:auxWeightsShape (Results 1 – 2 of 2) sorted by relevance
139 const Shape& auxWeightsShape, const T* recurrentWeightsData, in RNNStep() argument156 aux_input_weights_stride = auxWeightsShape.dimensions[1]; in RNNStep()233 const _Float16* auxWeightsData, const Shape& auxWeightsShape,249 const float* auxWeightsData, const Shape& auxWeightsShape,
65 const T* auxWeightsData, const Shape& auxWeightsShape,