Searched defs:sparse_tensor (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_sparse.py | 33 def bprop_sparse_tensor_get_indices(sparse_tensor, out, dout): argument
39 def bprop_sparse_tensor_get_values(sparse_tensor, out, dout): argument
47 def bprop_sparse_tensor_get_dense_shape(sparse_tensor, out, dout): argument
|
| /third_party/mindspore/mindspore/nn/sparse/ |
| D | sparse.py | 64 def construct(self, sparse_tensor): argument
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 373 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetValues() local
383 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetIndices() local
393 auto sparse_tensor = CheckArg<AbstractSparseTensor>(op_name, args_spec_list, 0); in InferImplSparseTensorGetDenseShape() local
|
| /third_party/mindspore/mindspore/common/ |
| D | dtype.py | 95 sparse_tensor = typing.SparseTensorType() variable
|
| /third_party/mindspore/third_party/proto/onnx/ |
| D | onnx.proto | 149 optional SparseTensorProto sparse_tensor = 22; // sparse tensor value field
|