Searched refs:kDataType (Results 1 – 2 of 2) sorted by relevance
35 auto data_type = TypeId(GetValue<int64_t>(primitive->GetAttr(kDataType))); in InferType()45 void ConstantOfShape::set_data_type(int64_t data_type) { (void)this->AddAttr(kDataType, MakeValue(d… in set_data_type()48 auto value_ptr = this->GetAttr(kDataType); in get_data_type()
59 constexpr auto kDataType = "data_type"; variable