Home
last modified time | relevance | path

Searched refs:kFloat64 (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/optimizer/
Dclean_test.cc57 …ractBasePtrList({std::make_shared<AbstractScalar>(kFloat64), std::make_shared<AbstractScalar>(kFlo… in SetUp()
62 AbstractBasePtr app_float = std::make_shared<AbstractScalar>(kFloat64); in SetUp()
63 AbstractBasePtr app_int = std::make_shared<AbstractScalar>(kFloat64); in SetUp()
65 …ractBasePtrList({std::make_shared<AbstractScalar>(kFloat64), std::make_shared<AbstractScalar>(kFlo… in SetUp()
107 std::vector<AbstractAttribute> attr = {{"x", std::make_shared<AbstractScalar>(kFloat64)}, in TEST_F()
108 {"y", std::make_shared<AbstractScalar>(kFloat64)}}; in TEST_F()
/third_party/flutter/engine/flutter/shell/platform/common/cpp/client_wrapper/
Dstandard_codec.cc35 kFloat64, enumerator
57 return EncodedType::kFloat64; in EncodedTypeForValue()
104 case EncodedType::kFloat64: { in ReadValue()
/third_party/mindspore/mindspore/core/ops/
Dfloor.cc35 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in InferType()
Dsigmoid_cross_entropy_with_logits.cc44 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SigmoidCrossEntropyWithLogitsInfer()
Dreverse_v2.cc36 kUInt32, kUInt64, kFloat16, kFloat32, kFloat64, kBool}; in InferType()
Delu.cc46 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in InferType()
Dzeros.cc49 kUInt16, kUInt32, kUInt64, kFloat16, kFloat32, kFloat64}; in ZerosInferType()
Dones.cc47 kUInt16, kUInt32, kUInt64, kFloat16, kFloat32, kFloat64}; in OnesInferType()
Dop_utils.h282 kUInt32, kUInt64, kFloat16, kFloat32, kFloat64};
286 kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64,
Dapply_momentum.cc79 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in ApplyMomentumInfer()
Dsoftmax.cc75 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in SoftMaxInferType()
Dindex_add.cc66 …onst std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kUInt8, kFloat16, kFloat32, kFloat64}; in IndexAddInferType()
Ddropout_do_mask.cc81 const std::set<TypePtr> keep_prop_valid_types = {kFloat16, kFloat32, kFloat64}; in InferType()
Dctcloss.cc101 const std::set<TypePtr> valid_types = {kFloat16, kFloat32, kFloat64}; in InferType()
Dmat_mul.cc95 …onst std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, kFloat64}; in MatMulInferType()
Dbatch_matmul.cc121 …onst std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, kFloat64}; in BatchMatmulInferType()
/third_party/mindspore/mindspore/core/ops/grad/
Dsigmoid_cross_entropy_with_logits_grad.cc47 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SigmoidCrossEntropyWithLogitsGradInfer()
Dsmooth_l1_loss_grad.cc54 … kUInt16, kUInt32, kUInt64, kFloat, kFloat16, kFloat32, kFloat64, kComplex64}; in SmoothL1LossGradInfer()
/third_party/mindspore/tests/ut/cpp/ir/
Dmeta_tensor_test.cc188 TensorPtr tensor = TensorPy::MakeTensor(py::array(tuple), kFloat64); in TEST_F()
224 TensorPtr tensor_float64 = TensorPy::MakeTensor(py::array(tuple), kFloat64); in TEST_F()
/third_party/mindspore/mindspore/core/ir/
Dscalar.h345 FP64Imm() : FloatImm(kFloat64), v_(0.0) {} in IMM_TRAITS()
346 …explicit FP64Imm(double v) : FloatImm(kFloat64), v_(v) { hash_ = hash_combine({tid(), std::hash<do… in IMM_TRAITS()
Ddtype_extends.cc66 …{kNumberTypeFloat64, kFloat64}, {kNumberTypeComplex64, kComplex64}, {kNumberTypeInt8, kInt8}, in TypeIdToType()
/third_party/mindspore/mindspore/core/ir/dtype/
Dnumber.h179 inline const TypePtr kFloat64 = std::make_shared<Float>(static_cast<int>(BitsNum::eBits64)); variable
/third_party/mindspore/tests/ut/cpp/session/
Danf_runtime_algorithm_test.cc601 …std::vector<TypeId> mutiple_types = {kFloat16->type_id(), kFloat32->type_id(), kFloat64->type_id()… in TEST_F()
606 EXPECT_EQ(AnfAlgo::GetOutputInferDataType(add, 2), kFloat64->type_id()); in TEST_F()
625 …std::vector<TypeId> mutiple_types = {kFloat16->type_id(), kFloat32->type_id(), kFloat64->type_id()… in TEST_F()
631 EXPECT_EQ(AnfAlgo::GetOutputInferDataType(first_add, 2), kFloat64->type_id()); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dprim_test.cc50 auto ele = std::make_shared<AbstractScalar>(kAnyValue, kFloat64); in ArrayFloat64Of()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc238 …(void)CheckTensorDType(x, {kFloat16, kFloat32, kFloat64, kInt32}, "Input 0 (x) for UnsortedSegment… in InferImplUnsortedSegmentSum()
1025 TypePtrList supported_types = {kInt64, kInt32, kFloat32, kFloat64}; in InferImplRange()

12