Searched refs:row_tensor (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 270 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_spec_list, 0); in InferImplRowTensorGetValues() local
271 MS_EXCEPTION_IF_NULL(row_tensor->values()); in InferImplRowTensorGetValues()
272 return row_tensor->values(); in InferImplRowTensorGetValues()
280 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_spec_list, 0); in InferImplRowTensorGetIndices() local
281 MS_EXCEPTION_IF_NULL(row_tensor->indices()); in InferImplRowTensorGetIndices()
282 return row_tensor->indices(); in InferImplRowTensorGetIndices()
290 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_spec_list, 0); in InferImplRowTensorGetDenseShape() local
291 MS_EXCEPTION_IF_NULL(row_tensor->dense_shape()); in InferImplRowTensorGetDenseShape()
292 return row_tensor->dense_shape(); in InferImplRowTensorGetDenseShape()
301 auto row_tensor = CheckArg<AbstractRowTensor>(op_name, args_spec_list, 0); in InferImplRowTensorAdd() local
[all …]
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_graphdata_distributed.py | 103 row_tensor = g.get_node_feature(nodes.tolist(), [1, 2, 3])
104 …assert row_tensor[0].tolist() == [[0, 1, 0, 0, 0], [1, 0, 0, 0, 1], [0, 0, 1, 1, 0], [0, 0, 0, 0, …
107 assert row_tensor[2].tolist() == [1, 2, 3, 1, 4, 3, 5, 3, 5, 4]
|
| D | test_graphdata.py | 37 row_tensor = g.get_node_feature(neighbor.tolist(), [2, 3])
38 assert row_tensor[0].shape == (10, 6)
|