Searched refs:tensor_slice (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_load_tensor.py | 27 tensor_slice = _load_tensor(tensor, dev_mat, tensor_map)
29 if expected_tensor.__str__() != tensor_slice.__str__():
33 tensor_slice = _load_tensor(tensor, dev_mat, tensor_map)
35 if expected_tensor.__str__() != tensor_slice.__str__():
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _tensor.py | 243 tensor_slice = _load_tensor(tensor, dev_mat, tensor_map)
248 tensor_slice = np.split(tensor_slice, size)[rank]
249 return Tensor(tensor_slice)
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 77 tensor_slice = P.Slice() variable
96 tensor_slice = P.Slice() variable
294 tensor_operator_registry.register('tensor_slice', tensor_slice)
|
| /third_party/mindspore/mindspore/numpy/ |
| D | utils.py | 182 return F.tensor_slice(f, index_start, index_end)
|
| D | array_ops.py | 1438 res = P.Split(axis, num_long_tensor)(F.tensor_slice(x, start1, size1)) + \
1439 P.Split(axis, num_short_tensor)(F.tensor_slice(x, start2, size2))
2069 cond_slice = F.tensor_slice(condlist.astype(mstype.float32), (i,) + slice_start, slice_size)
2070 choice_slice = F.tensor_slice(choicelist, (i,) + slice_start, slice_size)
|
| D | math_ops.py | 1966 minuend = F.tensor_slice(a, slice_start + (1,), slice_size)
1967 subtrahend = F.tensor_slice(a, slice_start + (0,), slice_size)
2079 y_slice_left = F.tensor_slice(y, y_start_axis_left + (0,) + y_start_axis_right, y_slice_size)
2082 y_slice_right = F.tensor_slice(y, y_start_axis_left + (1,) + y_start_axis_right, y_slice_size)
2971 return multiply(F.tensor_slice(a, a_slice_start + (idx_a,), a_slice_size),
2972 F.tensor_slice(b, b_slice_start + (idx_b,), b_slice_size))
3914 return F.tensor_slice(f, index_start, index_end)
|
| /third_party/mindspore/mindspore/_extends/parse/ |
| D | standard_method.py | 801 res = F.tensor_slice(res, begin, size)
|