| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | zeroslike_gpu_kernel.cc | 22 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTy…
25 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTy…
28 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberT…
31 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberT…
34 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumbe…
37 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumbe…
40 MS_REG_GPU_KERNEL_ONE(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumbe…
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | zeros_like.h | 29 class MS_CORE_API ZerosLike : public PrimitiveC {
32 ZerosLike() : PrimitiveC(prim::kPrimZerosLike->name()) { InitIOName({"x"}, {"y"}); } in ZerosLike() function
34 ~ZerosLike() = default;
35 MS_DECLARE_PARENT(ZerosLike, PrimitiveC);
41 using PrimZerosLikePtr = std::shared_ptr<ZerosLike>;
|
| D | zeros_like.cc | 52 REGISTER_PRIMITIVE_EVAL_IMPL(ZerosLike, prim::kPrimZerosLike, ZerosLikeInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_zeros_like_parser.cc | 32 auto prim = std::make_unique<ops::ZerosLike>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_zeros_like_parser.cc | 29 auto prim = std::make_unique<ops::ZerosLike>(); in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | elewise_calculation_ops_declare.h | 54 DECLARE_OP_ADAPTER(ZerosLike)
55 DECLARE_OP_USE_OUTPUT(ZerosLike)
|
| D | elewise_calculation_ops_declare.cc | 287 INPUT_MAP(ZerosLike) = {{1, INPUT_DESC(x)}};
288 ATTR_MAP(ZerosLike) = EMPTY_ATTR_MAP;
289 OUTPUT_MAP(ZerosLike) = {{0, OUTPUT_DESC(y)}};
290 REG_ADPT_DESC(ZerosLike, kNameZerosLike, ADPT_DESC(ZerosLike))
|
| /third_party/mindspore/mindspore/nn/probability/infer/variational/ |
| D | elbo.py | 55 self.zeros = P.ZerosLike()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.h | 68 MS_REG_CPU_KERNEL(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTyp…
70 MS_REG_CPU_KERNEL(ZerosLike, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeI…
|
| D | arithmetic_self_cpu_kernel.cc | 91 void ZerosLike(const T *, T *out, size_t size) { in ZerosLike() function
327 {prim::kPrimZerosLike->name(), ZerosLike<T>}, in LaunchKernel()
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_zeroslike_op.py | 28 self.zeros_like = P.ZerosLike()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_zeroslike_op.py | 28 self.zeros_like = P.ZerosLike()
81 self.zeros_like = P.ZerosLike()
|
| /third_party/mindspore/mindspore/boost/ |
| D | grad_accumulation.py | 41 zero_grad = P.ZerosLike()(cumulative_grad)
|
| /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/tests/st/armour/ |
| D | test_inversion_attack_grad.py | 56 self._zeros = P.ZerosLike()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 136 ZerosLike,
|
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | tinybert_train_export.py | 65 self.zeroslike = P.ZerosLike()
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 437 FUNC_MSOP2SCHEMAOP_DECLARE(ZerosLike)
|
| D | ops_def.cc | 188 OP_TYPE(ZerosLike)
1039 OP_SCHEMA_DEF(ZerosLike)
1040 OP_SCHEMA_DEF_END(ZerosLike)
|
| /third_party/mindspore/tests/st/probability/dpn/ |
| D | test_gpu_vae_gan.py | 113 self.zeros = ops.ZerosLike()
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_auto_grad.py | 500 self.zeros_like = P.ZerosLike()
531 self.zeros_like = P.ZerosLike()
562 self.zeros_like = P.ZerosLike()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 26 IsInstance, IsSubClass, ArgMaxWithValue, OnesLike, ZerosLike,
|
| /third_party/mindspore/mindspore/nn/probability/distribution/ |
| D | uniform.py | 169 self.zeroslike = P.ZerosLike()
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | base.py | 690 zeros_like = P.ZerosLike()
|
| /third_party/mindspore/tests/vm_impl/ |
| D | array_ops_vm_impl.py | 301 @vm_impl_getters.register(P.ZerosLike)
|