Searched refs:weights_buffer (Results 1 – 4 of 4) sorted by relevance
102 auto& weights_buffer = in Run() local104 const std::vector<float>& conv_weights_data = weights_buffer.data; in Run()117 weights_buffer.data = depthwise_conv_weights_data; in Run()
72 const auto& weights_buffer = weights.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine() local73 const float* const weights_data = weights_buffer.data.data(); in FuseAddOrSubParamsIntoFollowingAffine()
115 auto& weights_buffer = weights.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local116 float* const weights_data = weights_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine()
139 const auto weights_buffer = model.buffers[weights_tensor->buffer].get(); in TEST() local143 reinterpret_cast<int8_t*>(weights_buffer->data.data()); in TEST()146 ASSERT_EQ(sizeof(float) * weights_buffer->data.size(), in TEST()148 int num_values_in_channel = weights_buffer->data.size() / out_channel_size; in TEST()232 const auto weights_buffer = model.buffers[weights_tensor->buffer].get(); in TEST() local236 reinterpret_cast<int8_t*>(weights_buffer->data.data()); in TEST()239 ASSERT_EQ(sizeof(float) * weights_buffer->data.size(), in TEST()241 int num_values_in_channel = weights_buffer->data.size() / out_channel_size; in TEST()