Home
last modified time | relevance | path

Searched defs:dense_shape (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore/ops/_grad/
Dgrad_sparse.py27 def bprop_make_sparse_tensor(indices, values, dense_shape, out, dout): argument
56 def bprop(indices, values, dense_shape, out, dout): argument
69 def bprop(indices, values, dense_shape, dense, out, dout): argument
/third_party/mindspore/tests/ut/python/ir/
Dtest_sparse_tensor.py39 def __init__(self, dense_shape): argument
86 def __init__(self, dense_shape): argument
Dtest_row_tensor.py251 def __init__(self, dense_shape): argument
412 def __init__(self, dense_shape): argument
/third_party/mindspore/mindspore/core/ops/
Dsparse_to_dense.cc37 …auto dense_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->Build… in SparseToDenseInfer() local
/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_sparse_pynative.py46 def __init__(self, dense_shape): argument
Dtest_graph_param_cases.py99 def __init__(self, dense_shape): argument
/third_party/mindspore/mindspore/core/abstract/
Dabstract_value.h716 const AbstractTuplePtr dense_shape() const { return dense_shape_; } in dense_shape() function
717 void set_dense_shape(const AbstractTuplePtr &dense_shape) { dense_shape_ = dense_shape; } in set_dense_shape()
745 const AbstractTuplePtr dense_shape() const { return dense_shape_; } in dense_shape() function
746 void set_dense_shape(const AbstractTuplePtr &dense_shape) { dense_shape_ = dense_shape; } in set_dense_shape()
Dprim_others.cc210 auto dense_shape = CheckArg<AbstractTuple>(op_name, args_spec_list, 2); in InferImplMakeRowTensor() local
316 auto dense_shape = CheckArg<AbstractTuple>(op_name, args_spec_list, 2); in InferImplMakeSparseTensor() local
/third_party/mindspore/mindspore/common/
Dtensor.py1944 def __init__(self, indices, values, dense_shape): argument
1959 def dense_shape(self): member in RowTensor
2010 def __init__(self, indices, values, dense_shape): argument
2025 def dense_shape(self): member in SparseTensor
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py97 def __init__(self, dense_shape): argument