Searched refs:begin_params_axis_ (Results 1 – 15 of 15) sorted by relevance
46 param_->begin_params_axis_ = in ReSize()47 …param_->begin_params_axis_ > 0 ? param_->begin_params_axis_ : param_->begin_params_axis_ + shape.s… in ReSize()58 for (int i = 0; i < param_->begin_params_axis_; ++i) { in ReSize()62 for (size_t i = param_->begin_params_axis_; i < shape.size(); ++i) { in ReSize()
45 param_->begin_params_axis_ = in ReSize()46 …param_->begin_params_axis_ > 0 ? param_->begin_params_axis_ : param_->begin_params_axis_ + shape.s… in ReSize()57 for (int i = 0; i < param_->begin_params_axis_; ++i) { in ReSize()61 for (size_t i = param_->begin_params_axis_; i < shape.size(); ++i) { in ReSize()
112 param_->begin_params_axis_ = in ReSize()113 …param_->begin_params_axis_ > 0 ? param_->begin_params_axis_ : param_->begin_params_axis_ + shape.s… in ReSize()124 for (int i = 0; i < param_->begin_params_axis_; ++i) { in ReSize()128 for (size_t i = param_->begin_params_axis_; i < shape.size(); ++i) { in ReSize()
36 int begin_params_axis = param->begin_params_axis_; in LayerNormGradInferShape()37 if (param->begin_params_axis_ < 0) { in LayerNormGradInferShape()
42 if (param->begin_params_axis_ < (-1 * (int)(input->shape_size_)) || in LayerNormInferShape()43 param->begin_params_axis_ >= (int)(input->shape_size_)) { in LayerNormInferShape()
75 … __global FLT *beta_, int4 in_shape, float epsilon_, int begin_params_axis_) {89 if (begin_params_axis_ == 1) {92 } else if (begin_params_axis_ == 2) {95 } else if (begin_params_axis_ == 3) {
24 int begin_params_axis_; member
29 int begin_params_axis_; member
41 param->begin_params_axis_ = value->begin_params_axis(); in PopulateLayerNormGradParameter()
43 param->begin_params_axis_ = static_cast<int>(value->begin_params_axis()); in PopulateLayerNormParameter()
45 layer_norm_parameter->begin_params_axis_ = -static_cast<int>(normalized_shape->size()); in PopulateLayerNormParameterV0()
29 param->begin_params_axis_ = begin_param_axis_; in CreateParameter()
55 auto begin_params_axis = lngrad_param->begin_params_axis_; in Init()
64 auto begin_params_axis = lngrad_param->begin_params_axis_; in Init()
49 normalized_axis_ = param->begin_params_axis_; in CheckSpecs()