Searched refs:x_idx (Results 1 – 4 of 4) sorted by relevance
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_unique_op.py | 31 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/ |
D | winograd.cl | 54 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/ |
D | audio_utils.h | 205 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/ |
D | grad_nn_ops.py | 145 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)
|