Home
last modified time | relevance | path

Searched refs:x_idx (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/tests/st/ops/gpu/
Dtest_unique_op.py31 x_unique, x_idx = self.unique(x)
32 return x_unique, x_idx
44 x_unique, x_idx = self.unique(x_convert)
46 return x_unique, x_idx, x_split
58 x_unique, x_idx = net(x)
60 assert (x_idx.asnumpy() == exp_idx).all()
72 x_unique, x_idx = net(x)
74 assert (x_idx.asnumpy() == exp_idx).all()
86 x_unique, x_idx = net(x)
88 assert (x_idx.asnumpy() == exp_idx).all()
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/
Dwinograd.cl54 int x_idx = w * CI_SLICES + ci_slice;
57 …bt0 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 0)) + bt1 * READ_IMAGE(input, smp_zero, (int2)…
58 …bt2 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 2)) + bt3 * READ_IMAGE(input, smp_zero, (int2)…
59 …bt4 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 4)) + bt5 * READ_IMAGE(input, smp_zero, (int2)…
60 x_idx += CI_SLICES;
62 …bt0 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 0)) + bt1 * READ_IMAGE(input, smp_zero, (int2)…
63 …bt2 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 2)) + bt3 * READ_IMAGE(input, smp_zero, (int2)…
64 …bt4 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 4)) + bt5 * READ_IMAGE(input, smp_zero, (int2)…
65 x_idx += CI_SLICES;
67 …bt0 * READ_IMAGE(input, smp_zero, (int2)(x_idx, h + 0)) + bt1 * READ_IMAGE(input, smp_zero, (int2)…
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/audio/kernels/
Daudio_utils.h205 size_t x_idx = 0; in LFilter() local
222 while (x_idx < shape_1 * channel_idx) { in LFilter()
225 x_idx++; in LFilter()
235 for (size_t i = x_idx - shape_1; i < x_idx; i++) { in LFilter()
253 if (i + 1 == x_idx) continue; in LFilter()
262 if (x_idx % shape_1 == 0) { in LFilter()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_nn_ops.py145 x_idx = cast(F.tuple_to_array(range(1, x_indices_num)), mstype.float32)
146 x_idx = reshape(x_idx, (1, 1, x_row, x_col))
147 x_idx_patch = cast(extract_image_patches(x_idx), mstype.int32)