Searched refs:beta_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | BatchNormalizationLayerFixture.h | 60 void fill(U &&src_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U &&gamma_tensor) in fill() argument
71 library->fill(beta_tensor, distribution, 3); in fill()
76 library->fill_tensor_value(beta_tensor, 0.f); in fill()
|
| D | BatchNormalizationLayerFusionFixture.h | 66 …void fill(U &&src, U &&w_tensor, U &&b_tensor, U &&mean_tensor, U &&var_tensor, U &&beta_tensor, U… in fill() argument
76 …_use_beta ? library->fill(beta_tensor, distribution, 5) : library->fill_tensor_value(beta_tensor, … in fill()
|