Home
last modified time | relevance | path

Searched refs:ScalarToTensor (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/utils/
Dconvert_utils.h63 tensor::TensorPtr ScalarToTensor(const ScalarPtr &scalar);
Dconvert_utils.cc243 tensor::TensorPtr ScalarToTensor(const ScalarPtr &scalar) { in ScalarToTensor() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_scalar_to_tensor.cc39 tensor::TensorPtr tensor_ptr = ScalarToTensor(value->cast<ScalarPtr>()); in CreateTensorInput()
Dconvert_const_input_to_tensor_input.cc39 tensor_ptr = ScalarToTensor(value->cast<ScalarPtr>()); in CreateTensorInput()
/third_party/mindspore/mindspore/ops/operations/
D__init__.py29 … ScatterUpdate, ScalarToArray, ScalarToTensor, ScatterNd, ScatterNdUpdate, Select,
Darray_ops.py1514 class ScalarToTensor(PrimitiveWithInfer): class
/third_party/mindspore/mindspore/nn/probability/distribution/
Dlogistic.py141 self.consttensor = P.ScalarToTensor()
/third_party/mindspore/mindspore/ops/
Dfunctional.py126 scalar_to_tensor = P.ScalarToTensor()
/third_party/mindspore/mindspore/nn/layer/
Dmath.py702 self.const = P.ScalarToTensor()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc152 tensor::TensorPtr scalar_tensor = ScalarToTensor(value->cast<ScalarPtr>()); in PushInputTensor()