Searched defs:row_tensor (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/ |
| D | prim_others.cc | 190 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_abs_list, 0); in InferImplRowTensorGetValues() local
200 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_abs_list, 0); in InferImplRowTensorGetIndices() local
210 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_abs_list, 0); in InferImplRowTensorGetDenseShape() local
221 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_abs_list, 0); in InferImplRowTensorAdd() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | sparse_tensor.py | 39 def __init__(self, indices=None, values=None, shape=None, row_tensor=None): argument
127 def __init__(self, indices=None, values=None, shape=None, row_tensor=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 1103 py::tuple RowTensorPy::GetPyTupleShape(const RowTensor &row_tensor) { return GetSparseTensorShape(r… in GetPyTupleShape()
1113 ….def(py::init([](const RowTensor &row_tensor) { return std::make_shared<RowTensor>(row_tensor); }), in RegRowTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_guard/ |
| D | guard_utils.cc | 1436 …std::string row_tensor = DESC_TOSTRING(indices_) + DESC_TOSTRING(values_) + DESC_STRING(data_type_… in ToString() local
|