| /third_party/mindspore/mindspore/core/ops/ |
| D | rsqrt.h | 32 class MS_CORE_API Rsqrt : public PrimitiveC {
35 Rsqrt() : PrimitiveC(kNameRsqrt) { InitIOName({"x"}, {"output"}); } in Rsqrt() function
37 ~Rsqrt() = default;
38 MS_DECLARE_PARENT(Rsqrt, PrimitiveC);
44 using PrimRsqrtPtr = std::shared_ptr<Rsqrt>;
|
| D | rsqrt.cc | 53 REGISTER_PRIMITIVE_C(kNameRsqrt, Rsqrt);
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | lamb_next_mv_rule_test.py | 22 Rsqrt = P.Rsqrt() variable
52 sqrt0 = Rsqrt(add2)
87 sqrt0 = Rsqrt(add2)
111 sqrt0 = Rsqrt(add2)
135 sqrt0 = Rsqrt(add2)
159 sqrt0 = Rsqrt(add2)
187 sqrt0 = Rsqrt(add2)
221 sqrt0 = Rsqrt(add2)
250 sqrt0 = Rsqrt(add2)
284 sqrt0 = Rsqrt(add2)
[all …]
|
| D | lamb_next_mv_with_decay_rule_test.py | 22 Rsqrt = P.Rsqrt() variable
55 sqrt0 = Rsqrt(add2)
92 sqrt0 = Rsqrt(add2)
117 sqrt0 = Rsqrt(add2)
142 sqrt0 = Rsqrt(add2)
167 sqrt0 = Rsqrt(add2)
196 sqrt0 = Rsqrt(add2)
232 sqrt0 = Rsqrt(add2)
262 sqrt0 = Rsqrt(add2)
298 sqrt0 = Rsqrt(add2)
[all …]
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | arithmetic_self_npu.cc | 47 op_ = CreateOperator<hiai::op::Rsqrt>(name_); in Init()
105 SetInputs<hiai::op::Rsqrt>(npu_inputs, op_); in SetNPUInputs()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 85 DECLARE_OP_ADAPTER(Rsqrt)
86 DECLARE_OP_USE_OUTPUT(Rsqrt)
|
| D | elewise_calculation_ops_declare.cc | 443 INPUT_MAP(Rsqrt) = {{1, INPUT_DESC(x)}};
444 ATTR_MAP(Rsqrt) = EMPTY_ATTR_MAP;
445 OUTPUT_MAP(Rsqrt) = {{0, OUTPUT_DESC(y)}};
446 REG_ADPT_DESC(Rsqrt, kNameRsqrt, ADPT_DESC(Rsqrt))
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sqrt_op.py | 57 output_ms = P.Rsqrt()(Tensor(x_np))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | unary_op_gpu_kernel.cc | 71 MS_REG_GPU_KERNEL_ONE(Rsqrt, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTyp…
73 MS_REG_GPU_KERNEL_ONE(Rsqrt, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberTyp…
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 88 Rsqrt = 56, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unary_op_impl.cu | 499 void Rsqrt(const T *input, T *output, const size_t count, cudaStream_t cuda_stream) { in Rsqrt() function
567 template void Rsqrt<double>(const double *input, double *output, const size_t count, cudaStream_t c…
593 template void Rsqrt<float>(const float *input, float *output, const size_t count, cudaStream_t cuda…
618 template void Rsqrt<half>(const half *input, half *output, const size_t count, cudaStream_t cuda_st…
643 template void Rsqrt<char>(const char *input, char *output, const size_t count, cudaStream_t cuda_st…
685 template void Rsqrt<unsigned char>(const unsigned char *input, unsigned char *output, const size_t …
716 template void Rsqrt<int>(const int *input, int *output, const size_t count, cudaStream_t cuda_strea…
|
| D | unary_op_impl.cuh | 43 void Rsqrt(const T *input, T *output, const size_t count, cudaStream_t cuda_stream);
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 92 Rsqrt = 56, enumerator
|
| /third_party/spirv-tools/include/spirv/unified1/ |
| D | OpenCL.std.h | 92 Rsqrt = 56, enumerator
|