Home
last modified time | relevance | path

Searched defs:indptr (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/sparse/
Dtest_csr.py159 def construct(self, a, b, indptr, indices, values): argument
229 def construct(self, a, b, indptr, indices, values): argument
399 def construct(self, indptr, indices, values): argument
437 def construct(self, indptr, indices, values, dense_tensor, dense_vector): argument
568 def test_csr_mul(indptr, indices, values, shape, dense): argument
574 def test_csr_div(indptr, indices, values, shape, dense): argument
580 def test_csr_reduce_sum(indptr, indices, values, shape, axis): argument
586 def test_csrmv(indptr, indices, values, shape, dense): argument
701 def test_csr_tensor(indptr, indices, values, dense_shape): argument
707 def test_csr_indptr(indptr, indices, values, dense_shape): argument
[all …]
Dtest_sparse_dynshape.py70 def construct(self, indptr, indices, values, shape): argument
95 def construct(self, indptr, indices, values, dense_shape): argument
Dtest_control_flow.py201 def construct(self, indptr, indices, values, a, b): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/
Dgrad_sparse.py167 def bprop_make_csr_tensor(indptr, indices, values, dense_shape, out, dout): argument
200 def bprop(indptr, indices, values, shape, axis, out, dout): argument
213 def bprop(indptr, indices, values, dense_shape, dense, out, dout): argument
243 def bprop(indptr, indices, values, shape, dense, out, dout): argument
271 def bprop(indptr, indices, values, shape, dense, out, dout): argument
308 def bprop(indptr, nnz, out, dout): argument
328 def bprop_csr_sparse_matrix_to_dense(shape, batch, indptr, indices, values, out, dout): argument
/third_party/mindspore/mindspore-src/source/tests/st/gnn/gnn_csr/
Dtest_gcn_csr.py58 def construct(self, x, in_deg, out_deg, n_nodes, indptr, indices): argument
88 def construct(self, x, in_deg, out_deg, n_nodes, indptr, indices): argument
100 def construct(self, x, in_deg, out_deg, train_mask, target, n_nodes, indptr, indices): argument
Dgnngraph_dataset.py123 def construct(self, indptr, indices, values, shape, axis): argument
126 def bprop(self, indptr, indices, values, shape, axis, out, dout): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dcsr_to_coo.cc47 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, 0); in CSR2COOInfer() local
Dcsr_elementwise.cc45 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, 0); in CSRElementWiseInfer() local
Dcsr_gather.cc47 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, kIndexZero); in CSRGatherInfer() local
Dcsr_reducesum.cc49 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, 0); in CSRReduceSumInfer() local
Dcsr_mm.cc47 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, 0); in CSRMMInfer() local
Dcsr_mv.cc46 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, input_args, 0); in CSRMVInfer() local
Dmake_csrtensor.cc49 auto indptr = abstract::CheckArg<AbstractTensor>(op_name, args_spec_list, kIndexZero); in MakeCSRTensorInfer() local
/third_party/mindspore/mindspore-src/source/tests/st/ge/
Drun_sparse_tensor.py76 def construct(self, indptr, indices, values): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_sparse_ops.cc48 NodePtr CsrMulDiv(BpropBuilder *ib, const NodePtr &indptr, const NodePtr &indices, const NodePtr &v… in CsrMulDiv()
180 NodePtr BpropCSRMulDivCommon(BpropBuilder *ib, const NodePtr &indices, const NodePtr &indptr, const… in BpropCSRMulDivCommon()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dsparse_matrix_softmax_impl.cu25 __global__ void SparseMatrixSoftmaxKernel(int rows, IndexType *indptr, DataType *values, DataType *… in SparseMatrixSoftmaxKernel()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/
Dsparse_tensor.py710 def __init__(self, indptr=None, indices=None, values=None, shape=None, csr_tensor=None): argument
773 def indptr(self) -> Tensor: member in CSRTensor
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dtensor.cc493 CSRTensor::CSRTensor(const TensorPtr indptr, const TensorPtr indices, const TensorPtr values, const… in CSRTensor()
517 auto indptr = indptr_->ToAbstract()->cast<abstract::AbstractTensorPtr>(); in ToAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dconvert_utils_py.cc723 TensorPtr indptr = utils::cast<TensorPtr>(value_sequeue[tensor::CSRTensor::kIndptrIdx]); in MakeCSRTensor() local
737 TensorPtr indptr = utils::cast<TensorPtr>(value_list[tensor::CSRTensor::kIndptrIdx]); in MakeCSRTensor() local
Dconvert_utils.cc615 …tensor::TensorPtr indptr = utils::cast<tensor::TensorPtr>(tensor_list[tensor::CSRTensor::kIndptrId… in TensorListToCSRTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/
Demitter.h208 …NodePtr CSR2COO(const NodePtr &indptr, const NodePtr &nnz) { return Emit("CSR2COO", {indptr, nnz})… in CSR2COO()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_py.cc1067 …[](const BaseTensor &indptr, const BaseTensor &indices, const BaseTensor &values, const py::tuple … in RegCSRTensor()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/
D_checkparam.py1049 def check_csr_tensor_input(indptr, indices, values, shape): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/
Dmindir_exporter.cc654 mind_ir::AttributeProto *indptr = attr_proto->add_values(); in SetCSRTensorToProto() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dabstract_value.cc2662 const AbstractTensorPtr AbstractCSRTensor::indptr() const { in indptr() function in mindspore::abstract::AbstractCSRTensor