Searched refs:kKeepDims (Results 1 – 7 of 7) sorted by relevance
25 void All::set_keep_dims(const int64_t keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(keep_di… in set_keep_dims()28 auto value_ptr = GetAttr(kKeepDims); in get_keep_dims()
75 auto keep_dims = GetValue<bool>(primitive->GetAttr(kKeepDims)); in InferShape()90 void Reduce::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(keep_di… in set_keep_dims()92 bool Reduce::get_keep_dims() const { return GetValue<bool>(GetAttr(kKeepDims)); } in get_keep_dims()
89 auto keep_dimis_value_ptr = primitive->GetAttr(kKeepDims); in InferShape()
104 constexpr auto kKeepDims = "keep_dims"; variable
28 void ArgMaxFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()35 auto keep_dims = GetAttr(kKeepDims); in get_keep_dims()
28 void ArgMinFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()33 auto keep_dims = GetAttr(kKeepDims); in get_keep_dims()
29 void ReduceFusion::set_keep_dims(const bool keep_dims) { (void)this->AddAttr(kKeepDims, MakeValue(k… in set_keep_dims()43 auto value_ptr = GetAttr(kKeepDims); in get_keep_dims()