| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | oneslike_gpu_kernel.cc | 19 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumber…
21 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumber…
23 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumber…
25 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTyp…
27 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTy…
29 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTy…
31 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTy…
33 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTy…
35 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberT…
37 MS_REG_GPU_KERNEL_ONE(OnesLike, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberT…
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | ones_like.h | 29 class MS_CORE_API OnesLike : public PrimitiveC {
32 OnesLike() : PrimitiveC(prim::kPrimOnesLike->name()) {} in OnesLike() function
34 ~OnesLike() = default;
35 MS_DECLARE_PARENT(OnesLike, PrimitiveC);
41 using PrimOnesLikePtr = std::shared_ptr<OnesLike>;
|
| D | ones_like.cc | 52 REGISTER_PRIMITIVE_EVAL_IMPL(OnesLike, prim::kPrimOnesLike, OnesLikeInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | oneslike_impl.cu | 21 __global__ void OnesLike(const size_t size, const T* input, T* output) { in OnesLike() function
31 OnesLike<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, input, output); in CalOnesLike()
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
| D | oneslike.py | 19 class OnesLike(Expander): class
|
| D | __init__.py | 45 from .oneslike import OnesLike
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 57 DECLARE_OP_ADAPTER(OnesLike)
58 DECLARE_OP_USE_OUTPUT(OnesLike)
|
| D | elewise_calculation_ops_declare.cc | 293 INPUT_MAP(OnesLike) = {{1, INPUT_DESC(x)}};
294 ATTR_MAP(OnesLike) = EMPTY_ATTR_MAP;
295 OUTPUT_MAP(OnesLike) = {{0, OUTPUT_DESC(y)}};
296 REG_ADPT_DESC(OnesLike, kNameOnesLike, ADPT_DESC(OnesLike))
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.h | 72 MS_REG_CPU_KERNEL(OnesLike, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberType…
74 MS_REG_CPU_KERNEL(OnesLike, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeIn…
|
| D | arithmetic_self_cpu_kernel.cc | 81 void OnesLike(const T *, T *out, size_t size) { in OnesLike() function
328 {prim::kPrimOnesLike->name(), OnesLike<T>}, in LaunchKernel()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_oneslike_op.py | 28 self.ones_like = P.OnesLike()
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_oneslike.py | 27 self.oneslike = P.OnesLike()
|
| /third_party/mindspore/mindspore/nn/probability/bijector/ |
| D | scalar_affine.py | 90 self.oneslike = P.OnesLike()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_oneslike_op.py | 30 self.ones_like = P.OnesLike()
|
| /third_party/mindspore/mindspore/ops/bprop_mindir/ |
| D | generate_mindir.py | 25 serializable_bprop_ops = [P.ReLU(), P.Identity(), inner.Range(1.0), P.OnesLike(), P.ZerosLike(), P.…
|
| /third_party/mindspore/mindspore/nn/grad/ |
| D | cell_grad.py | 74 self.oneslike = P.OnesLike()
|
| /third_party/mindspore/mindspore/nn/loss/ |
| D | loss.py | 860 self.ones_like = P.OnesLike()
1048 self.ones = P.OnesLike()
1214 self.ones = P.OnesLike()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 236 OnesLike,
|
| D | ops.fbs | 126 OnesLike,
761 table OnesLike {
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 368 FUNC_MSOP2SCHEMAOP_DECLARE(OnesLike)
|
| D | ops_def.cc | 125 OP_TYPE(OnesLike)
761 OP_SCHEMA_DEF(OnesLike)
762 OP_SCHEMA_DEF_END(OnesLike)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 26 IsInstance, IsSubClass, ArgMaxWithValue, OnesLike, ZerosLike,
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 42 ones_like = P.OnesLike()
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | adafactor.py | 134 update_coff = P.Maximum()(update_rms_thres, P.OnesLike()(update_rms_thres))
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 689 @bprop_getters.register(P.OnesLike)
956 ones_like = P.OnesLike()
|