Searched defs:GatherNd (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | gather_nd.h | 32 GatherNd() : PrimitiveC(kNameGatherNd) { InitIOName({"input_x", "indices"}, {"y"}); } in GatherNd() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | gatherNd_fp32.c | 21 int GatherNd(const float *input, float *output, const int *in_offset, int area, int count) { in GatherNd() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | gathernd.cu | 49 void GatherNd(T *input, S *indices, T *output, const size_t &output_dim0, const size_t &output_dim1, in GatherNd() function
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 3718 class GatherNd(PrimitiveWithInfer): class
|