Home
last modified time | relevance | path

Searched refs:Sqrt (Results 1 – 25 of 204) sorted by relevance

123456789

/third_party/mindspore/mindspore/core/ops/
Dsqrt.h27 class MS_CORE_API Sqrt : public PrimitiveC {
30 Sqrt() : PrimitiveC(kNameSqrt) { InitIOName({"x"}, {"output"}); } in Sqrt() function
32 ~Sqrt() = default;
33 MS_DECLARE_PARENT(Sqrt, PrimitiveC);
Dsqrt.cc24 REGISTER_PRIMITIVE_C(kNameSqrt, Sqrt);
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dadam_apply_one_fusion_test.py25 Sqrt = P.Sqrt() variable
57 sqrt0 = Sqrt(add1)
75 sqrt0 = Sqrt(add1)
93 sqrt0 = Sqrt(add1)
111 sqrt0 = Sqrt(add1)
129 sqrt0 = Sqrt(add1)
161 sqrt0 = Sqrt(add1)
185 sqrt0 = Sqrt(add1)
209 sqrt0 = Sqrt(add1)
233 sqrt0 = Sqrt(add1)
[all …]
Dlamb_next_right_rule_test.py21 Sqrt = P.Sqrt() variable
49 sqrt0 = Sqrt(real_div1)
62 sqrt0 = Sqrt(real_div1)
Dlamb_next_mv_rule_test.py23 Sqrt = P.Sqrt() variable
53 sqrt1 = Sqrt(real_div1)
88 sqrt1 = Sqrt(real_div1)
112 sqrt1 = Sqrt(real_div1)
136 sqrt1 = Sqrt(real_div1)
160 sqrt1 = Sqrt(real_div1)
188 sqrt1 = Sqrt(real_div1)
222 sqrt1 = Sqrt(real_div1)
251 sqrt1 = Sqrt(real_div1)
285 sqrt1 = Sqrt(real_div1)
[all …]
Dlamb_next_mv_with_decay_rule_test.py23 Sqrt = P.Sqrt() variable
52 sqrt1 = Sqrt(real_div1)
89 sqrt1 = Sqrt(real_div1)
114 sqrt1 = Sqrt(real_div1)
139 sqrt1 = Sqrt(real_div1)
164 sqrt1 = Sqrt(real_div1)
193 sqrt1 = Sqrt(real_div1)
229 sqrt1 = Sqrt(real_div1)
259 sqrt1 = Sqrt(real_div1)
295 sqrt1 = Sqrt(real_div1)
[all …]
Dclip_by_norm_no_div_square_sum_fusion.py21 sqrt = P.Sqrt()
/third_party/skia/tests/sksl/intrinsics/
DSqrt.asm.frag138 %34 = OpExtInst %float %1 Sqrt %40
149 %48 = OpExtInst %v2float %1 Sqrt %51
164 %61 = OpExtInst %v3float %1 Sqrt %64
178 %75 = OpExtInst %v4float %1 Sqrt %77
231 %123 = OpExtInst %float %1 Sqrt %float_n1
242 %131 = OpExtInst %v2float %1 Sqrt %132
254 %141 = OpExtInst %v3float %1 Sqrt %142
267 %151 = OpExtInst %v4float %1 Sqrt %152
/third_party/skia/third_party/externals/swiftshader/src/Reactor/
DOptimalIntrinsics.cpp109 …return As<Float4>(As<Int4>(half_pi - Sqrt(Float4(1.0f) - absx) * (a0 + absx * (a1 + absx * (a2 + a… in Asin_4_terms()
127 …return As<Float4>(As<Int4>(half_pi - Sqrt(Float4(1.0f) - absx) * (a0 + absx * (a1 + absx * (a2 + a… in Asin_8_terms()
284 return optimal::Log(x + Sqrt(x * x + Float4(1.0f))); in Asinh()
289 return optimal::Log(x + Sqrt(x + Float4(1.0f)) * Sqrt(x - Float4(1.0f))); in Acosh()
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_recompute.py41 self.sqrt = P.Sqrt()
57 self.sqrt = P.Sqrt()
71 self.sqrt = P.Sqrt()
86 self.sqrt = P.Sqrt()
/third_party/grpc/examples/csharp/RouteGuide/RouteGuide/
DRouteGuideUtil.cs72 double c = 2 * Math.Atan2(Math.Sqrt(a), Math.Sqrt(1 - a)); in GetDistance()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dunary_op_gpu_kernel.cc65 MS_REG_GPU_KERNEL_ONE(Sqrt, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberType…
67 MS_REG_GPU_KERNEL_ONE(Sqrt, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberType…
69 MS_REG_GPU_KERNEL_ONE(Sqrt, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberType…
/third_party/flutter/skia/tests/
DSkVxTest.cpp33 float2 Sqrt(float2 x) { return sqrt(x); } in Sqrt() function
34 float4 Sqrt(float4 x) { return sqrt(x); } in Sqrt() function
35 float8 Sqrt(float8 x) { return sqrt(x); } in Sqrt() function
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Darithmetic_self_npu.cc44 op_ = CreateOperator<hiai::op::Sqrt>(name_); in Init()
102 SetInputs<hiai::op::Sqrt>(npu_inputs, op_); in SetNPUInputs()
/third_party/mindspore/mindspore/nn/optim/
Dadafactor.py39 rel_step_sz = P.Minimum()(min_step, 1.0 / P.Sqrt()(step * 1.0))
54 div_val = 1.0 / P.Sqrt()(P.Div()(exp_avg_sq_row, reduce_mean))
58 c_factor = 1.0 / P.Sqrt()(exp_avg_sq_col)
130 exp_avg_sq_update = 1.0 / P.Sqrt()(exp_avg_sq_update)
/third_party/json/benchmarks/thirdparty/benchmark/src/
Dstatistics.cc60 auto Sqrt = [](const double dat) { in __anonb80b20840402() variable
75 return Sqrt(v.size() / (v.size() - 1.0) * (avg_squares - Sqr(mean))); in StatisticsStdDev()
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_sqrt.py28 self.sqrt = P.Sqrt()
/third_party/benchmark/src/
Dstatistics.cc61 auto Sqrt = [](const double dat) { in __anon693a7bb40402() variable
75 return Sqrt(v.size() / (v.size() - 1.0) * (avg_squares - Sqr(mean))); in StatisticsStdDev()
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Delewise_calculation_ops_declare.h88 DECLARE_OP_ADAPTER(Sqrt)
89 DECLARE_OP_USE_OUTPUT(Sqrt)
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_sqrt.cc33 auto sqrt = std::make_shared<Sqrt>(); in TEST_F()
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/optimizer/
Dopt_test.py1064 Sqrt = Primitive('Sqrt')
1072 return Mul(tensor1, Mul(tensor2, Sqrt(x)))
1076 return Mul(tensor1, Mul(Sqrt(x), tensor2))
1080 return Mul(Mul(Sqrt(x), tensor2), tensor1)
1084 return Mul(Mul(Sqrt(x), tensor2), tensor1)
1088 return Mul(Sqrt(x), Mul(tensor1, tensor2))
/third_party/mindspore/tests/st/ops/gpu/
Dtest_sqrt_op.py28 output_ms = P.Sqrt()(Tensor(x_np))
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/
Deltwise_cpu_kernel.h56 MS_REG_CPU_KERNEL(Sqrt, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloa…
/third_party/skia/third_party/externals/swiftshader/third_party/subzero/src/
DIceIntrinsics.h53 Sqrt, enumerator
/third_party/skia/third_party/externals/spirv-headers/include/spirv/1.2/
DOpenCL.std.h93 Sqrt = 61, enumerator

123456789