Searched refs:GetDefaultDtype (Results 1 – 11 of 11) sorted by relevance
| /dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/routines/ |
| H A D | type_util.cc | 40 scalar_max_dtype_ = PromoteTypes(*scalar_max_dtype_, internal::GetDefaultDtype(arg.kind())); in AddArg()
42 scalar_max_dtype_ = internal::GetDefaultDtype(arg.kind()); in AddArg()
|
| H A D | type_util_test.cc | 17 TEST(DtypeTest, GetDefaultDtype) { in TEST() argument
18 EXPECT_EQ(Dtype::kBool, internal::GetDefaultDtype(DtypeKind::kBool)); in TEST()
19 EXPECT_EQ(Dtype::kInt32, internal::GetDefaultDtype(DtypeKind::kInt)); in TEST()
20 EXPECT_EQ(Dtype::kFloat32, internal::GetDefaultDtype(DtypeKind::kFloat)); in TEST()
|
| H A D | type_util.h | 16 inline Dtype GetDefaultDtype(DtypeKind kind) { in GetDefaultDtype() function
84 inline Dtype ResultType(Scalar arg) { return internal::GetDefaultDtype(arg.kind()); } in ResultType()
|
| H A D | creation.cc | 105 return Full(shape, fill_value, internal::GetDefaultDtype(fill_value.kind()), device); in Full()
138 return Arange(start, stop, step, internal::GetDefaultDtype(step.kind()), device); in Arange()
145 return Arange(start, stop, 1, internal::GetDefaultDtype(stop.kind()), device); in Arange()
150 Array Arange(Scalar stop, Device& device) { return Arange(0, stop, 1, internal::GetDefaultDtype(sto… in Arange()
319 Dtype dtype_a = dtype.value_or(internal::GetDefaultDtype(chainerx::DtypeKind::kFloat)); in Linspace()
|
| H A D | statistics.cc | 111 …pe) { return GetKind(dtype) == DtypeKind::kFloat ? dtype : internal::GetDefaultDtype(DtypeKind::kF… in PromoteInt2Float()
|
| H A D | arithmetic.cc | 398 dtype = internal::GetDefaultDtype(DtypeKind::kFloat); in TrueDivide()
406 dtype = internal::GetDefaultDtype(DtypeKind::kFloat); in TrueDivide()
414 dtype = internal::GetDefaultDtype(DtypeKind::kFloat); in TrueDivide()
|
| H A D | creation_test.cc | 105 EXPECT_EQ(x.dtype(), internal::GetDefaultDtype(scalar.kind())); in CheckFullWithScalarDtype()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
| H A D | np_true_divide-inl.h | 61 CHECK_EQ(out.type_flag_, mxnet::common::GetDefaultDtype()) in TrueDivideScalarCompute()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
| H A D | np_true_divide-inl.h | 61 CHECK_EQ(out.type_flag_, mxnet::common::GetDefaultDtype()) in TrueDivideScalarCompute()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/common/ |
| H A D | utils.h | 918 inline int GetDefaultDtype() {
924 inline int GetDefaultDtype(int dtype) {
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/common/ |
| H A D | utils.h | 918 inline int GetDefaultDtype() {
924 inline int GetDefaultDtype(int dtype) {
|