Home
last modified time | relevance | path

Searched defs:CSRTensor (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Dsparse_unary_func.py23 def csr_cos(x: CSRTensor) -> CSRTensor:
104 def csr_tan(x: CSRTensor) -> CSRTensor:
177 def csr_exp(x: CSRTensor) -> CSRTensor:
250 def csr_inv(x: CSRTensor) -> CSRTensor:
323 def csr_relu(x: CSRTensor) -> CSRTensor:
406 def csr_expm1(x: CSRTensor) -> CSRTensor:
481 def csr_isfinite(x: CSRTensor) -> CSRTensor:
560 def csr_asin(x: CSRTensor) -> CSRTensor:
638 def csr_sqrt(x: CSRTensor) -> CSRTensor:
711 def csr_log(x: CSRTensor) -> CSRTensor:
[all …]
Dsparse_func.py169 def csr_div(x: CSRTensor, y: Tensor) -> Tensor:
203 def csr_mul(x: CSRTensor, y: Tensor) -> CSRTensor:
228 def csr_mv(csr_tensor: CSRTensor, dense: Tensor) -> Tensor:
246 def csr_mm(a: CSRTensor, b: CSRTensor, trans_a: bool = False, trans_b: bool = False,
323 def csr_reduce_sum(csr_tensor: CSRTensor, axis: int) -> Tensor:
340 def csr_to_coo(tensor: CSRTensor) -> COOTensor:
382 def csr_to_dense(csr_tensor: CSRTensor) -> Tensor:
727 def csr_softmax(logits: CSRTensor, dtype: mstype):
774 def csr_add(a: CSRTensor, b: CSRTensor, alpha: Tensor, beta: Tensor) -> CSRTensor:
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/
Dsparse_tensor.py639 class CSRTensor(CSRTensor_): class
1045 def mm(self, matrix: Union[Tensor, CSRTensor]) -> Union[Tensor, CSRTensor]: argument
1138 def add(self, b: CSRTensor, alpha: Tensor, beta: Tensor) -> CSRTensor:
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dtensor.h585 MS_DECLARE_PARENT(CSRTensor, MetaSparseTensor) in MS_DECLARE_PARENT() argument
Dtensor.cc493 CSRTensor::CSRTensor(const TensorPtr indptr, const TensorPtr indices, const TensorPtr values, const… in CSRTensor() function in mindspore::tensor::CSRTensor
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils_py.cc708 using CSRTensor = tensor::CSRTensor; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Ddata_converter.cc54 using CSRTensor = mindspore::tensor::CSRTensor; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dpipeline.cc130 using CSRTensor = mindspore::tensor::CSRTensor; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/
Dguard_utils.cc122 CSRTensor, enumerator