| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | gatherv2.cu | 42 void GatherV2(T *input, S *indices, T *output, size_t output_dim0, size_t output_dim1, size_t outpu… in GatherV2() function
50 template void GatherV2<float, int>(float *input, int *indices, float *output, size_t output_dim0, s…
52 template void GatherV2<float, int64_t>(float *input, int64_t *indices, float *output, size_t output…
54 template void GatherV2<half, int>(half *input, int *indices, half *output, size_t output_dim0, size…
56 template void GatherV2<half, int64_t>(half *input, int64_t *indices, half *output, size_t output_di…
58 template void GatherV2<double, int>(double *input, int *indices, double *output, size_t output_dim0…
60 template void GatherV2<double, int64_t>(double *input, int64_t *indices, double *output, size_t out…
62 template void GatherV2<int, int>(int *input, int *indices, int *output, size_t output_dim0, size_t …
64 template void GatherV2<int, int64_t>(int *input, int64_t *indices, int *output, size_t output_dim0,…
66 template void GatherV2<int16_t, int>(int16_t *input, int *indices, int16_t *output, size_t output_d…
[all …]
|
| D | gatherv2.cuh | 21 void GatherV2(T *input, S *indices, T *output, size_t output_dim0, size_t output_dim1, size_t outpu…
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gather_v2_primitive.py | 66 class GatherV2(LossBase): class
68 super(GatherV2, self).__init__()
158 …criterion = GatherV2(1, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
165 …criterion = GatherV2(2, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
172 …criterion = GatherV2(1, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
180 …criterion = GatherV2(1, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
187 …criterion = GatherV2(1, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
194 …criterion = GatherV2(1, strategy=gather_v2_strategy, index_size=batch_size_per_device * device_num…
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | selection_ops_declare.h | 104 DECLARE_OP_ADAPTER(GatherV2)
105 DECLARE_OP_USE_OUTPUT(GatherV2)
|
| D | selection_ops_declare.cc | 30 INPUT_MAP(GatherV2) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(indices)}, {3, INPUT_DESC(axis)}};
31 ATTR_MAP(GatherV2) = EMPTY_ATTR_MAP;
32 OUTPUT_MAP(GatherV2) = {{0, OUTPUT_DESC(y)}};
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gatherv2_gpu_kernel.h | 62 GatherV2(input_addr, indices_addr, output_addr, dims_[0], dims_[1], dims_[2], input_dim1, in Launch()
|
| /third_party/mindspore/tests/ut/python/pipeline/infer/ |
| D | test_auto_monad.py | 112 self.gather = P.GatherV2()
326 self.gather = P.GatherV2()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 25 … Fill, Ones, Zeros, GatherNd, GatherV2, Gather, SparseGatherV2, InvertPermutation,
|
| D | array_ops.py | 841 class GatherV2(PrimitiveWithCheck): class
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 215 self.gather = P.GatherV2()
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 426 self.gather = P.GatherV2()
742 self.gather = P.GatherV2()
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | transformer.py | 528 self.gather = P.GatherV2().shard(((1, 1), (parallel_config.data_parallel, 1)))
534 self.gather = P.GatherV2().shard(((parallel_config.model_parallel, 1), (1, 1)))
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | thor_layer.py | 591 self.gather = P.GatherV2()
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 406 @bprop_getters.register(P.GatherV2)
|
| /third_party/mindspore/ |
| D | RELEASE.md | 1489 ##### `ops.GatherV2`, change API name to `ops.Gather` ([!11713](https://gitee.com/mindspore/mindspo…
1491 GatherV2 is changed to Gather. The old interface can be used continuously, but will be deleted in s…
1503 >>> gather = ops.GatherV2()
|