Searched refs:FloatImm (Results 1 – 8 of 8) sorted by relevance
308 class MS_CORE_API FloatImm : public Scalar {310 FloatImm() = default;311 explicit FloatImm(const TypePtr &t) : Scalar(t) {} in FloatImm() function312 ~FloatImm() override = default;313 MS_DECLARE_PARENT(FloatImm, Scalar)315 using FloatImmPtr = std::shared_ptr<FloatImm>;317 class MS_CORE_API FP32Imm : public FloatImm {319 FP32Imm() : FloatImm(kFloat32), v_(0.0) {} in FP32Imm()320 …explicit FP32Imm(float v) : FloatImm(kFloat32), v_(v) { hash_ = hash_combine({tid(), std::hash<flo… in FP32Imm()322 MS_DECLARE_PARENT(FP32Imm, FloatImm) in MS_DECLARE_PARENT() argument[all …]
107 } else if (means->isa<FloatImm>()) { in Init()119 } else if (stds->isa<FloatImm>()) { in Init()
108 } else if (means->isa<FloatImm>()) { in Init()121 } else if (stds->isa<FloatImm>()) { in Init()
34 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("means")->isa<FloatImm>()) { in InitKernel()46 } else if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("stds")->isa<FloatImm>()) { in InitKernel()
98 if (!keep_prop_value->isa<FloatImm>()) { in InferType()
40 if (scalar->isa<FloatImm>()) { in IsNodeScalarTrueWith()
244 } else if (scalar->isa<FloatImm>()) { in CreateTupleTensor()