| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gatherv2_gpu_kernel.cc | 22 Gather,
26 Gather,
30 Gather,
34 Gather,
38 Gather,
42 Gather,
46 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
49 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt64).AddOutputAttr(k…
52 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt16).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
55 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt16).AddInputAttr(kNumberTypeInt64).AddOutputAttr(k…
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | gather_cpu_kernel.h | 49 …Gather, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
52 …Gather, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
55 …Gather, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
58 …Gather, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddInputAttr(kNumberTypeInt32).AddOutputAttr(…
61 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt8).AddInputAttr(kNumberTypeInt32).AddOutputAttr(kN…
64 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt16).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
67 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt32).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
70 …Gather, KernelAttr().AddInputAttr(kNumberTypeInt64).AddInputAttr(kNumberTypeInt32).AddOutputAttr(k…
73 Gather,
77 Gather,
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | gather.cu | 44 void Gather(const T *input, const S *index, T *output, const size_t dim_before_axis, in Gather() function
53 template void Gather<double, int>(const double *input, const int *index, double *output,
57 template void Gather<double, int64_t>(const double *input, const int64_t *index, double *output,
61 template void Gather<float, int>(const float *input, const int *index, float *output,
65 template void Gather<float, int64_t>(const float *input, const int64_t *index, float *output,
69 template void Gather<half, int>(const half *input, const int *index, half *output,
73 template void Gather<half, int64_t>(const half *input, const int64_t *index, half *output,
77 template void Gather<int64_t, int>(const int64_t *input, const int *index, int64_t *output,
81 template void Gather<int64_t, int64_t>(const int64_t *input, const int64_t *index, int64_t *output,
85 template void Gather<int, int>(const int *input, const int *index, int *output,
[all …]
|
| D | gather.cuh | 20 void Gather(const T *input, const S *index, T *output, const size_t dim_before_axis,
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | gather.h | 32 class MS_CORE_API Gather : public PrimitiveC {
35 Gather() : PrimitiveC(kNameGather) { InitIOName({"param", "indices", "axis"}, {"output"}); } in Gather() function
37 ~Gather() = default;
38 MS_DECLARE_PARENT(Gather, PrimitiveC);
42 using PrimGatherPtr = std::shared_ptr<Gather>;
|
| /third_party/glslang/Test/ |
| D | hlsl.gather.array.dx10.frag | 29 float4 txval20 = g_tTex2df4a . Gather(g_sSamp, float3(0.1, 0.2, 0.3));
30 int4 txval21 = g_tTex2di4a . Gather(g_sSamp, float3(0.3, 0.4, 0.5));
31 uint4 txval22 = g_tTex2du4a . Gather(g_sSamp, float3(0.5, 0.6, 0.7));
35 float4 txval40 = g_tTexcdf4a . Gather(g_sSamp, float4(0.1, 0.2, 0.3, 0.4));
36 int4 txval41 = g_tTexcdi4a . Gather(g_sSamp, float4(0.4, 0.5, 0.6, 0.7));
37 uint4 txval42 = g_tTexcdu4a . Gather(g_sSamp, float4(0.7, 0.8, 0.9, 1.0));
|
| D | hlsl.gather.basic.dx10.frag | 34 float4 txval20 = g_tTex2df4 . Gather(g_sSamp, float2(0.1, 0.2));
35 int4 txval21 = g_tTex2di4 . Gather(g_sSamp, float2(0.3, 0.4));
36 uint4 txval22 = g_tTex2du4 . Gather(g_sSamp, float2(0.5, 0.6));
40 float4 txval40 = g_tTexcdf4 . Gather(g_sSamp, float3(0.1, 0.2, 0.3));
41 int4 txval41 = g_tTexcdi4 . Gather(g_sSamp, float3(0.4, 0.5, 0.6));
42 uint4 txval42 = g_tTexcdu4 . Gather(g_sSamp, float3(0.7, 0.8, 0.9));
|
| D | hlsl.gather.basic.dx10.vert | 33 float4 txval20 = g_tTex2df4 . Gather(g_sSamp, float2(0.1, 0.2));
34 int4 txval21 = g_tTex2di4 . Gather(g_sSamp, float2(0.3, 0.4));
35 uint4 txval22 = g_tTex2du4 . Gather(g_sSamp, float2(0.5, 0.6));
39 float4 txval40 = g_tTexcdf4 . Gather(g_sSamp, float3(0.1, 0.2, 0.3));
40 int4 txval41 = g_tTexcdi4 . Gather(g_sSamp, float3(0.4, 0.5, 0.6));
41 uint4 txval42 = g_tTexcdu4 . Gather(g_sSamp, float3(0.7, 0.8, 0.9));
|
| D | hlsl.gather.offsetarray.dx10.frag | 25 float4 txval20 = g_tTex2df4 . Gather(g_sSamp, float3(0.1, 0.2, 0.3), int2(1,0));
26 int4 txval21 = g_tTex2di4 . Gather(g_sSamp, float3(0.3, 0.4, 0.4), int2(1,1));
27 uint4 txval22 = g_tTex2du4 . Gather(g_sSamp, float3(0.5, 0.6, 0.7), int2(1,-1));
|
| D | hlsl.gather.offset.dx10.frag | 33 float4 txval20 = g_tTex2df4 . Gather(g_sSamp, float2(0.1, 0.2), int2(1,0));
34 int4 txval21 = g_tTex2di4 . Gather(g_sSamp, float2(0.3, 0.4), int2(1,1));
35 uint4 txval22 = g_tTex2du4 . Gather(g_sSamp, float2(0.5, 0.6), int2(1,-1));
|
| D | remap.hlsl.sample.basic.strip.frag | 35 int Gather; // ...
59 mtest.Gather = 1; // ...
|
| D | remap.hlsl.sample.basic.everything.frag | 35 int Gather; // ...
59 mtest.Gather = 1; // ...
|
| D | remap.hlsl.sample.basic.none.frag | 35 int Gather; // ...
59 mtest.Gather = 1; // ...
|
| D | hlsl.sample.basic.dx10.frag | 35 int Gather; // ...
59 mtest.Gather = 1; // ...
|
| /third_party/vk-gl-cts/external/openglcts/modules/gles31/ |
| D | es31cTextureGatherTests.cpp | 796 Offset() + " mediump " + Type() + " tmp = " + Gather() + in TextBody()
805 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::GatherBase
1074 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherUint2D
1114 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherDepth2D
1133 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherFloat2DArray
1172 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherInt2DArray
1211 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherUint2DArray
1251 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherDepth2DArray
1275 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherFloatCube
1299 virtual std::string Gather() in Gather() function in glcts::__anon7ca765f00111::PlainGatherIntCube
[all …]
|
| /third_party/boost/libs/asio/doc/overview/ |
| D | implementation.qbk | 32 Scatter-Gather:
52 Scatter-Gather:
73 Scatter-Gather:
95 Scatter-Gather:
116 Scatter-Gather:
137 Scatter-Gather:
159 Scatter-Gather:
181 Scatter-Gather:
203 Scatter-Gather:
224 Scatter-Gather:
[all …]
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_dynamic_shape.py | 59 self.embedding_lookp = P.Gather().shard(((1, 8), (1,)))
62 self.gatherv2 = P.Gather().shard(((1, 8), (1,)))
93 self.embedding_lookp = P.Gather().shard(((8, 1), (1,)))
96 self.gatherv2 = P.Gather().shard(((1, 1), (1,)))
|
| /third_party/glslang/Test/baseResults/ |
| D | hlsl.sample.basic.dx10.frag.out | 13 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
20 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
26 0:59 Gather: direct index for structure ( temp int)
27 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
34 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
41 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
48 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
55 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
62 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
69 …ulateLevelOfDetail, temp int CalculateLevelOfDetailUnclamped, temp int Gather, temp int GetDime…
[all …]
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_gather_op.py | 29 self.gatherv2 = P.Gather()
55 self.gatherv2 = P.Gather()
81 self.gatherv2 = P.Gather()
|
| /third_party/skia/third_party/externals/swiftshader/src/Reactor/ |
| D | EmulatedIntrinsics.hpp | 27 RValue<Float4> Gather(RValue<Pointer<Float>> base, RValue<Int4> offsets, RValue<Int4> mask, unsigne…
28 RValue<Int4> Gather(RValue<Pointer<Int>> base, RValue<Int4> offsets, RValue<Int4> mask, unsigned in…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | gather_base.h | 27 int Gather(const void *input, int outer_size, int inner_size, int limit, const int *indices, int in…
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_gatherV2_op.py | 29 self.gather = P.Gather()
853 self.gather = P.Gather()
907 self.gather = P.Gather()
947 self.gather = P.Gather()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_gather_parser.cc | 25 auto prim = std::make_unique<ops::Gather>(); in Parse()
|
| /third_party/python/Tools/peg_generator/pegen/ |
| D | metagrammar.gram | 8 Gather,
107 | sep=atom '.' node=atom '+' {Gather(sep, node)}
|
| D | first_sets.py | 12 Gather,
95 def visit_Gather(self, item: Gather) -> Set[str]: argument
|