Searched defs:num_units (Results 1 – 9 of 9) sorted by relevance
86 const uint32_t num_units = num_filters / rank; in Prepare() local173 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
188 uint32_t num_units() const { return units_; } in num_units() function in android::nn::wrapper::BasicRNNOpModel
294 int num_units() const { return units_; } in num_units() function in android::nn::wrapper::SVDFOpModel
201 uint32_t num_units = getSizeOfDimension(weights, 0); in validateShapes() local
47 num_units = 16 variable
50 num_units = 16 variable
24 num_units = 4 variable