Home
last modified time | relevance | path

Searched refs:ScalarToArray (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/nn/probability/dpn/vae/
Dvae.py66 self.to_tensor = P.ScalarToArray()
Dcvae.py73 self.to_tensor = P.ScalarToArray()
/third_party/mindspore/mindspore/ops/composite/
Darray_ops.py187 to_tensor_op = P.ScalarToArray()
/third_party/mindspore/mindspore/nn/probability/distribution/
Dlog_normal.py90 self.const = P.ScalarToArray()
Dgumbel.py90 self.const = P.ScalarToArray()
Dbernoulli.py147 self.const = P.ScalarToArray()
Dexponential.py150 self.const = P.ScalarToArray()
Dgeometric.py155 self.const = P.ScalarToArray()
Dnormal.py161 self.const = P.ScalarToArray()
Dcauchy.py158 self.const = P.ScalarToArray()
Duniform.py160 self.const = P.ScalarToArray()
Dlogistic.py140 self.const = P.ScalarToArray()
/third_party/mindspore/tests/st/probability/dpn/
Dtest_gpu_vae_gan.py97 self.to_tensor = ops.ScalarToArray()
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/
Droi_align.py172 mask = self.equal(target_lvls, P.ScalarToArray()(i))
/third_party/mindspore/mindspore/ops/operations/
D__init__.py29 … ScatterUpdate, ScalarToArray, ScalarToTensor, ScatterNd, ScatterNdUpdate, Select,
Darray_ops.py1473 class ScalarToArray(PrimitiveWithInfer): class
/third_party/mindspore/mindspore/nn/probability/toolbox/
Duncertainty_evaluation.py340 self.to_tensor = P.ScalarToArray()
/third_party/mindspore/mindspore/ops/
Dfunctional.py125 scalar_to_array = P.ScalarToArray()
/third_party/mindspore/mindspore/nn/layer/
Dmath.py614 self.const = P.ScalarToArray()