| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | unary_op_gpu_kernel.cc | 117 MS_REG_GPU_KERNEL_ONE(Rint, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberType…
119 MS_REG_GPU_KERNEL_ONE(Rint, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberType…
121 MS_REG_GPU_KERNEL_ONE(Rint, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberType…
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 124 DECLARE_OP_ADAPTER(Rint)
125 DECLARE_OP_USE_OUTPUT(Rint)
|
| D | elewise_calculation_ops_declare.cc | 327 INPUT_MAP(Rint) = {{1, INPUT_DESC(x)}};
328 ATTR_MAP(Rint) = EMPTY_ATTR_MAP;
329 OUTPUT_MAP(Rint) = {{0, OUTPUT_DESC(y)}};
330 REG_ADPT_DESC(Rint, kNameRint, ADPT_DESC(Rint))
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_rint_op.py | 27 self.rint = ops.Rint()
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/spirv-headers/include/spirv/1.1/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/flutter/skia/third_party/externals/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/spirv-headers/include/spirv/1.2/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/spirv-headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/1.0/ |
| D | OpenCL.std.h | 85 Rint = 53, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | unary_op_impl.cu | 534 void Rint(const T *input, T *output, const size_t count, cudaStream_t cuda_stream) { in Rint() function
570 template void Rint<double>(const double *input, double *output, const size_t count, cudaStream_t cu…
596 template void Rint<float>(const float *input, float *output, const size_t count, cudaStream_t cuda_…
621 template void Rint<half>(const half *input, half *output, const size_t count, cudaStream_t cuda_str…
646 template void Rint<char>(const char *input, char *output, const size_t count, cudaStream_t cuda_str…
691 template void Rint<unsigned char>(const unsigned char *input, unsigned char *output, const size_t c…
719 template void Rint<int>(const int *input, int *output, const size_t count, cudaStream_t cuda_stream…
|
| D | unary_op_impl.cuh | 63 void Rint(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 | 89 Rint = 53, enumerator
|
| /third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 89 Rint = 53, enumerator
|
| /third_party/spirv-tools/include/spirv/unified1/ |
| D | OpenCL.std.h | 89 Rint = 53, enumerator
|
| /third_party/mesa3d/src/compiler/spirv/ |
| D | OpenCL.std.h | 89 Rint = 53, enumerator
|
| /third_party/spirv-headers/include/spirv/unified1/ |
| D | OpenCL.std.h | 89 Rint = 53, enumerator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.cc | 111 void Rint(const T *in, T *out, size_t size) { in Rint() function
330 {prim::kPrimRint->name(), Rint<T>}, in LaunchKernel()
|
| D | arithmetic_self_cpu_kernel.h | 82 MS_REG_CPU_KERNEL(Rint, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeFloa…
|