Searched refs:layer_size_ (Results 1 – 3 of 3) sorted by relevance
39 float *out_data = output_data + num * parameter->layer_size_; in CopyData()40 float *in_data = input_data + ids[num] * parameter->layer_size_; in CopyData()42 l2_regulate(in_data, parameter->layer_size_, parameter->max_norm_); in CopyData()46 memcpy(out_data, in_data, sizeof(float) * (size_t)(parameter->layer_size_)); in CopyData()
30 int layer_size_; member
39 param_->layer_size_ = 1; in ReSize()42 param_->layer_size_ *= in_shape[i]; in ReSize()79 …reinterpret_cast<float *>(ms_context_->allocator->Malloc(sizeof(float) * param_->layer_size_ * par… in Run()