Home
last modified time | relevance | path

Searched defs:dtype_value (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Darange.cc30 bool CheckDtypeValidAndIsInteger(const PrimitivePtr &primitive, const ValuePtr &dtype_value) { in CheckDtypeValidAndIsInteger()
118 auto dtype_value = input_args[kInputIndex3]->GetValue(); in InferShape() local
158 const auto &dtype_value = input_values[kInputIndex3]; in InferType() local
174 const auto &dtype_value = input_values[kInputIndex3]; in InferShape() local
Dscalar_to_tensor.cc56 auto dtype_value = GetScalarValue<int64_t>(input_args[kInputIndex1]->GetValue()); in InferType() local
Dtuple_to_tensor.cc93 auto dtype_value = GetScalarValue<int64_t>(input_args[kInputIndex1]->GetValue()); in InferType() local
Ddropout_gen_mask_ext.cc72 auto dtype_value = static_cast<TypeId>(dtype_opt.value()); in CheckValidation() local
Deye.cc98 auto dtype_value = TypeIdToType(static_cast<TypeId>(dtype_ptr.value())); in CheckValidation() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_ones.cc30 ValuePtr dtype_value; member
48 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
76 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
Dtest_ops_zeros.cc30 ValuePtr dtype_value; member
48 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
76 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
Dtest_ops_fill.cc32 ValuePtr dtype_value; member
51 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
81 auto dtype_value = param.dtype_value->ToAbstract(); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Ddropout_gen_mask_ext.cc43 auto dtype_value = static_cast<TypeId>(dtype->value()); in DropoutGenMaskExtAscendCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dnon_deterministic_ints.cc81 auto dtype_value = prim->GetAttr("dtype"); in NonDeterministicIntsInferType() local
Drandom_poisson.cc98 auto dtype_value = prim->GetAttr("dtype"); in RandomPoissonInferType() local
Drandom_categorical.cc104 auto dtype_value = prim->GetAttr("dtype"); in RandomCategoricalInferType() local
Dtruncated_normal.cc105 auto dtype_value = prim->GetAttr("dtype"); in TruncatedNormalInferType() local
Dhamming_window.cc121 int64_t dtype_value = GetValue<int64_t>(dtype_attr); in HammingWindowInferType() local
Dcrop_and_resize_grad_image.cc147 auto dtype_value = primitive->GetAttr("T"); in CropAndResizeGradImageInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dtuple_to_tensor.cc161 auto dtype_value = GetScalarValue<int64_t>(input_args[kInputIndex1]->GetValue()); in InferValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_for_ge.cc126 auto dtype_value = MakeValue(TypeIdToType(dtype_id)); in GenCastNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/debugger/
Dtensor_summary.cc178 void TensorSummary<T>::TensorStatistics(DbgDataType dtype_value) { in TensorStatistics()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Ddropout_unify_mindir.cc266 auto dtype_value = static_cast<int64_t>(GetInputXDataType(dropout)); in CreateDropoutGenMaskCNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/common/
Dgllo_utils.cc1194 auto dtype_value = MakeValue(dst_type_ptr); in GenCastNode() local