Searched refs:SparseGatherV2 (Results 1 – 19 of 19) sorted by relevance
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_sparse_gather_v2_op.py | 28 self.gather = P.SparseGatherV2()
693 self.gather = P.SparseGatherV2()
742 self.gather = P.SparseGatherV2()
780 self.gather = P.SparseGatherV2()
821 self.gather = P.SparseGatherV2()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gatherv2_gpu_kernel.cc | 140 SparseGatherV2,
144 SparseGatherV2,
147 MS_REG_GPU_KERNEL_TWO(SparseGatherV2,
154 MS_REG_GPU_KERNEL_TWO(SparseGatherV2,
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 274 class SparseGatherV2(nn.Cell): class
276 super(SparseGatherV2, self).__init__()
283 GradWrap(SparseGatherV2())(params, indices)
298 class SparseGatherV2(nn.Cell): class
300 super(SparseGatherV2, self).__init__()
307 network = GradWrap(SparseGatherV2())
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_sparse_feature_bprop.py | 58 self.gatherv2 = P.SparseGatherV2()
117 self.gatherv2 = P.SparseGatherV2()
|
| D | test_sparse_gather_v2.py | 55 … self.gatherv2 = P.SparseGatherV2().shard(strategy1).add_prim_attr("primitive_target", target)
|
| /third_party/mindspore/tests/st/pynative/dynamic_shape/ |
| D | test_pynative_ftrl.py | 33 self.gather = P.SparseGatherV2()
|
| /third_party/mindspore/tests/ut/python/nn/optim/ |
| D | test_ftrl.py | 53 self.gather = P.SparseGatherV2()
|
| D | test_proximal_ada_grad.py | 53 self.gather = P.SparseGatherV2()
|
| D | test_adam.py | 64 self.gather = P.SparseGatherV2()
|
| D | test_adafactor.py | 67 self.gather = P.SparseGatherV2()
|
| /third_party/mindspore/tests/st/dynamic_shape/ |
| D | test_ftrl.py | 33 self.gather = P.SparseGatherV2()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 25 … Fill, Ones, Zeros, GatherNd, GatherV2, Gather, SparseGatherV2, InvertPermutation,
|
| D | array_ops.py | 863 class SparseGatherV2(PrimitiveWithCheck): class
|
| /third_party/mindspore/tests/st/hcom/ |
| D | hcom_sparsetensor.py | 135 self.gather = P.SparseGatherV2()
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 240 self.gatherv2 = P.SparseGatherV2()
|
| D | thor_layer.py | 739 self.gatherv2 = P.SparseGatherV2()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 483 @bprop_getters.register(P.SparseGatherV2)
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 1934 'block': P.SparseGatherV2(),
|
| /third_party/mindspore/config/ |
| D | op_info.config | 467 {"op_name": "SparseGatherV2", "inputs": [{"index": 0, "name": "x", "need_compile": false, "param_ty…
468 {"op_name": "SparseGatherV2", "inputs": [{"index": 0, "name": "x", "need_compile": false, "param_ty…
|