Searched defs:num_units (Results 1 – 6 of 6) sorted by relevance
69 const uint32_t num_units = num_filters / rank; in Prepare() local157 const int num_units = num_filters / rank; in EvalFloat32() local
66 const uint32_t num_units = SizeOfDimension(input_weights, 0); in Prepare() local150 const uint32_t num_units = weightsShape.dimensions[0]; in RNNStep() local
212 uint32_t num_units() const { return units_; } in num_units() function in android::nn::wrapper::BasicRNNOpModel
211 uint32_t num_units = getSizeOfDimension(weights, 0); in prepare() local
318 int num_units() const { return units_; } in num_units() function in android::nn::wrapper::SVDFOpModel
47 num_units = 16 variable