Searched refs:gather_nd (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 52 gather_nd = P.GatherNd()
56 update_grad = neg(gather_nd(dout, indices))
64 gather_nd = P.GatherNd()
70 possibly_updated = gather_nd(out, indices)
75 …dupdates = gather_nd(dout / F.cast(indicators, F.dtype(dout)), indices) * F.cast(out_indicators, F…
|
| /third_party/mindspore/mindspore/ops/_op_impl/cpu/ |
| D | __init__.py | 36 from .gather_nd import _gather_nd_cpu
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_sparse.py | 57 return zeros_like(indices), F.gather_nd(dout, indices), zeros_like(dense_shape)
|
| D | grad_array_ops.py | 771 gather_nd = P.GatherNd()
776 update_grad = gather_nd(dout, indices)
785 gather_nd = P.GatherNd()
788 update_grad = gather_nd(dout, indices)
|
| /third_party/mindspore/mindspore/ops/ |
| D | functional.py | 137 gather_nd = P.GatherNd() variable
297 tensor_operator_registry.register('gather_nd', gather_nd)
|
| /third_party/mindspore/mindspore/numpy/ |
| D | math_ops.py | 4331 a = F.gather_nd(a, sorter)
4339 mask = less_op(v, F.gather_nd(a, F.expand_dims(mid, -1)))
4401 x_0 = F.gather_nd(xp, indices_0)
4402 x_1 = F.gather_nd(xp, indices_1)
4403 y_0 = F.gather_nd(fp, indices_0)
4404 y_1 = F.gather_nd(fp, indices_1)
4411 left = F.gather_nd(fp, idx_0)
4414 right = F.gather_nd(fp, idx_last)
4416 res = F.select(F.tensor_lt(x, F.gather_nd(xp, idx_0)), left, res)
4417 res = F.select(F.tensor_gt(x, F.gather_nd(xp, idx_last)), right, res)
|
| D | array_ops.py | 2186 return F.gather_nd(choices, indices)
|
| /third_party/mindspore/mindspore/_extends/parse/ |
| D | standard_method.py | 982 return F.gather_nd(choices, indices).astype(dtype)
1024 a = F.gather_nd(a, sorter)
1032 mask = less_op(v, F.gather_nd(a, mid.reshape(mid.shape + (1,))))
|
| /third_party/mindspore/mindspore/ops/_op_impl/tbe/ |
| D | __init__.py | 114 from .gather_nd import _gather_nd_tbe
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _compile_utils.py | 396 result = F.gather_nd(data, indices)
471 result = F.gather_nd(data, indices)
|
| /third_party/mindspore/config/ |
| D | op_info.config | 243 … "TBE", "async_flag": false, "binfile_name": "gather_nd.so", "compute_cost": 10, "kernel_name": "g…
|