Home
last modified time | relevance | path

Searched refs:RowTensor (Results 1 – 15 of 15) sorted by relevance

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_graph_param_cases.py17 from mindspore import RowTensor
86 return RowTensor(indices, values, dense_shape)
96 return RowTensor(indices, values, dense_shape)
107 x = RowTensor(indices, values, self.dense_shape)
Dtest_sparse_pynative.py24 from mindspore import context, Tensor, RowTensor, SparseTensor
50 x = RowTensor(indices, values, self.dense_shape)
/third_party/mindspore/mindspore/nn/wrap/
Dgrad_reducer.py19 from mindspore.common.tensor import RowTensor
179 grad = RowTensor(indices, dout, grad.dense_shape)
209 grad = RowTensor(indices, dout, grad.dense_shape)
275 return RowTensor(grad.indices, dout, grad.dense_shape)
Dloss_scale.py21 from ...common import Tensor, RowTensor
39 return RowTensor(grad.indices,
/third_party/mindspore/tests/st/pynative/
Dtest_graph_param_transform.py17 from mindspore import RowTensor
86 return RowTensor(indices, values, dense_shape)
94 return RowTensor(indices, values, dense_shape)
105 x = RowTensor(indices, values, self.dense_shape)
/third_party/mindspore/tests/ut/python/ir/
Dtest_row_tensor.py33 from mindspore import Tensor, RowTensor, context
140 return RowTensor(indices, values, x_shp), zeros_like(indices), zeros_like(axis)
243 ret = (RowTensor(indices, values, self.dense_shape),)
255 x = RowTensor(indices, values, self.dense_shape)
397 return RowTensor(indices, values, dense_shape)
408 return RowTensor(indices, values, dense_shape)
419 x = RowTensor(indices, values, self.dense_shape)
/third_party/mindspore/mindspore/common/
D__init__.py20 from .tensor import Tensor, RowTensor, SparseTensor
Dtensor.py1895 class RowTensor: class
/third_party/mindspore/mindspore/_extends/parse/
Dresources.py21 from mindspore import RowTensor, SparseTensor
140 RowTensor: F.make_row_tensor,
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_comm_ops.py23 from ...common.tensor import RowTensor
64 dx = RowTensor(indices, grad, dout.dense_shape)
80 dx = RowTensor(indices, grad, dout.dense_shape)
430 dx = RowTensor(indices, grad, dout.dense_shape)
437 dx = RowTensor(indices, grad, dout.dense_shape)
Dgrad_array_ops.py31 from ...common.tensor import RowTensor
113 return RowTensor(dout.indices, values, dout.dense_shape)
288 return RowTensor(new_indices, actual_dout, x_shp), zeros_like(indices), zeros_like(offset)
498 return RowTensor(indices_new, values, x_shp), zeros_like(indices), zeros_like(axis)
/third_party/mindspore/mindspore/nn/optim/
Doptimizer.py28 from mindspore.common.tensor import Tensor, RowTensor
669 return RowTensor(indices, values, shape)
695 return RowTensor(indices, values, shape)
735 …return RowTensor(grad.indices, grad.values * F.cast(scale, F.dtype(grad.values)), grad.dense_shape)
747 return RowTensor(unique_indices, summed_values, grad.dense_shape)
/third_party/mindspore/mindspore/core/abstract/
Dparam_validator.h84 ABSTRACT_REPORT_NAME_TRAITS(RowTensor) in ABSTRACT_REPORT_NAME_TRAITS()
/third_party/mindspore/mindspore/boost/
Dboost_cell_wrapper.py65 return RowTensor(grad.indices,
/third_party/mindspore/
DRELEASE.md2623 … - change IndexedSlices to RowTensor([!4031](https://gitee.com/mindspore/mindspore/pulls/4031))