Searched refs:kNumberTypeUInt64 (Results 1 – 25 of 89) sorted by relevance
1234
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reshape_cpu_kernel.h | 59 …_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
78 …_REG_CPU_KERNEL(Flatten, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
99 …_CPU_KERNEL(FlattenGrad, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
120 …G_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
139 …_REG_CPU_KERNEL(Squeeze, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | scatter_nd_cpu_kernel.h | 86 …Attr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
131 …Attr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | assign_cpu_kernel.h | 78 …Assign, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddInputAttr(kNumberTypeUInt64).AddOutputAttr…
|
| D | gather_cpu_kernel.h | 58 …Gather, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
111 .AddInputAttr(kNumberTypeUInt64)
114 .AddOutputAttr(kNumberTypeUInt64),
|
| D | spacetodepth_cpu_kernel.h | 79 …KernelAttr().SetAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUInt64),
|
| D | depthtospace_cpu_kernel.h | 80 …KernelAttr().SetAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUInt64),
|
| D | unpack_cpu_kernel.h | 74 …KernelAttr().SetAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUInt64),
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | cast_gpu_kernel.cc | 35 …_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeUInt64), CastGpuKer…
66 …TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTypeUInt64), CastGpuKer…
95 …TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeUInt64), CastGpuKer…
124 …TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeUInt64), CastGpuKer…
153 …TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeUInt64), CastGpuKer…
182 …GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTypeUInt64),
211 …GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTypeUInt64),
226 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeI…
228 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeI…
230 MS_REG_GPU_KERNEL_TWO(Cast, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeI…
[all …]
|
| D | gather_gpu_kernel.cc | 94 …GatherD, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
97 …GatherD, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddInputAttr(kNumberTypeInt64).AddOutputAttr…
|
| D | batchtospace_gpu_kernel.cc | 39 …KERNEL_ONE(BatchToSpace, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | spacetobatch_gpu_kernel.cc | 39 …KERNEL_ONE(SpaceToBatch, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | depthtospace_gpu_kernel.cc | 38 …KERNEL_ONE(DepthToSpace, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | spacetodepth_gpu_kernel.cc | 38 …KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | oneslike_gpu_kernel.cc | 39 …GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | pack_gpu_kernel.cc | 46 …ck, KernelAttr().AddAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | meshgrid_gpu_kernel.cc | 45 …id, KernelAttr().AddAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | strided_slice_grad_gpu_kernel.cc | 35 …EL_ONE(StridedSliceGrad, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | unpack_gpu_kernel.cc | 46 …ck, KernelAttr().AddAllSameAttr(true).AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| D | strided_slice_gpu_kernel.cc | 35 …KERNEL_ONE(StridedSlice, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | push_kernel.cc | 31 .AddOutputAttr(kNumberTypeUInt64),
35 …).AddInputAttr(kNumberTypeFloat32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeUInt64),
43 .AddOutputAttr(kNumberTypeUInt64),
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | fed_avg_kernel.cc | 26 .AddInputNameType(kDataSize, kNumberTypeUInt64)
28 .AddInputNameType(kNewDataSize, kNumberTypeUInt64),
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | identity_gpu_kernel.cc | 27 …GPU_KERNEL_ONE(Identity, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUIn…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/other/ |
| D | gpu_convert_to_dynamic_shape_gpu_kernel.cc | 67 KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUInt64),
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_convert_utils.cc | 43 {"uint64", TypeId::kNumberTypeUInt64},
65 {TypeId::kNumberTypeUInt64, "uint64"},
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | type_id.h | 42 case kNumberTypeUInt64: in MSTypeToDEType()
76 return mindspore::TypeId::kNumberTypeUInt64; in DETypeToMSType()
|
1234