| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gathernd_gpu_kernel.cc | 22 GatherNd,
26 GatherNd,
30 GatherNd,
34 …GatherNd, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
37 …GatherNd, KernelAttr().AddInputAttr(kNumberTypeInt16).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
40 GatherNd,
44 …GatherNd, KernelAttr().AddInputAttr(kNumberTypeInt8).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
47 …GatherNd, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddInputAttr(kNumberTypeInt32).AddOutputAttr…
50 …GatherNd, KernelAttr().AddInputAttr(kNumberTypeBool).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
53 GatherNd,
[all …]
|
| D | gathernd_gpu_kernel.h | 69 … GatherNd(input_addr, indices_addr, output_addr, dims_[0], dims_[1], dims_[2], dev_batch_strides_, in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | gathernd_cpu_kernel.h | 50 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, bool);
51 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int8_t);
52 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int16_t);
53 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int32_t);
54 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, int64_t);
55 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint8_t);
56 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint16_t);
57 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint32_t);
58 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, uint64_t);
59 MS_REG_CPU_KERNEL_T(GatherNd, KernelAttr(), GatherNdCPUKernel, float);
[all …]
|
| /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
57 template void GatherNd<double, int>(double *input, int *indices, double *output, const size_t &outp…
60 template void GatherNd<float, int>(float *input, int *indices, float *output, const size_t &output_…
63 template void GatherNd<half, int>(half *input, int *indices, half *output, const size_t &output_dim…
66 template void GatherNd<int, int>(int *input, int *indices, int *output, const size_t &output_dim0,
69 template void GatherNd<short, int>(short *input, int *indices, short *output, const size_t &output_…
72 template void GatherNd<unsigned int, int>(unsigned int *input, int *indices, unsigned int *output,
76 template void GatherNd<char, int>(char *input, int *indices, char *output, const size_t &output_dim…
79 template void GatherNd<unsigned char, int>(unsigned char *input, int *indices, unsigned char *outpu…
83 template void GatherNd<bool, int>(bool *input, int *indices, bool *output, const size_t &output_dim…
[all …]
|
| D | gathernd.cuh | 23 void GatherNd(T *input, S *indices, T *output, const size_t &output_dim0, const size_t &output_dim1,
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | gather_nd.h | 29 class MS_CORE_API GatherNd : public PrimitiveC {
32 GatherNd() : PrimitiveC(kNameGatherNd) { InitIOName({"input_x", "indices"}, {"y"}); } in GatherNd() function
34 ~GatherNd() = default;
35 MS_DECLARE_PARENT(GatherNd, PrimitiveC);
41 using PrimGatherNd = std::shared_ptr<GatherNd>;
|
| D | gather_nd.cc | 67 REGISTER_PRIMITIVE_C(kNameGatherNd, GatherNd);
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_gathernd_op.py | 41 op = P.GatherNd()
56 op = P.GatherNd()
71 op = P.GatherNd()
87 op = P.GatherNd()
103 op = P.GatherNd()
119 op = P.GatherNd()
135 op = P.GatherNd()
151 op = P.GatherNd()
167 op = P.GatherNd()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | selection_ops_declare.cc | 97 INPUT_MAP(GatherNd) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(indices)}};
98 ATTR_MAP(GatherNd) = EMPTY_ATTR_MAP;
99 OUTPUT_MAP(GatherNd) = {{0, OUTPUT_DESC(y)}};
100 REG_ADPT_DESC(GatherNd, kNameGatherNd, ADPT_DESC(GatherNd))
|
| D | selection_ops_declare.h | 37 DECLARE_OP_ADAPTER(GatherNd)
38 DECLARE_OP_USE_OUTPUT(GatherNd)
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 52 gather_nd = P.GatherNd()
64 gather_nd = P.GatherNd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | gatherNd_fp32.h | 30 int GatherNd(const float *input, float *output, const int *in_offset, int area, int count);
|
| 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/lite/tools/converter/parser/tf/ |
| D | tf_gather_nd_parser.cc | 29 auto prim = std::make_unique<ops::GatherNd>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_gather_nd_parser.cc | 31 auto prim = std::make_unique<ops::GatherNd>(); in Parse()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_gathernd_further.py | 53 self.gathernd = P.GatherNd().shard(strategy2)
69 self.gathernd = P.GatherNd().shard(strategy2)
84 self.gathernd = P.GatherNd().shard(strategy2)
|
| D | test_gathernd.py | 53 self.gathernd = P.GatherNd().shard(strategy2)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | gather_nd_infer.c | 56 REG_INFER(GatherNd, PrimType_GatherNd, GatherNdInferShape)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 723 @bprop_getters.register(P.GatherNd)
738 op = P.GatherNd()
749 op = P.GatherNd()
760 op = P.GatherNd()
771 gather_nd = P.GatherNd()
785 gather_nd = P.GatherNd()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | gatherNd_fp32.cc | 111 auto ret = GatherNd(in_ptr_, out_ptr_ + offset * area_, in_offset_ + offset, area_, count); in DoGatherNd()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 144 GatherNd,
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 329 FUNC_MSOP2SCHEMAOP_DECLARE(GatherNd)
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_gathernd_op.py | 26 self.gathernd = P.GatherNd()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 25 … Fill, Ones, Zeros, GatherNd, GatherV2, Gather, SparseGatherV2, InvertPermutation,
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
| D | proposal_generator.py | 97 self.gatherND = P.GatherNd()
|