| /third_party/mindspore/tests/st/pynative/ |
| D | test_tensor_getitem.py | 167 def construct(self, x, index_0, index_1, index_2): argument
170 ret2 = x[index_0, index_1, index_2] + self.const2
185 index_2 = np.random.randint(6, size=(5, 3, 4, 5)).astype(np.int32)
189 input_2_ms = Tensor(index_2)
193 assert np.all(output2.asnumpy() == input_x[index_0, index_1, index_2] + np.ones([5, 3, 4, 5]))
356 def construct(self, index_0, index_1, index_2): argument
372 index_2 = np.random.randint(3, size=(2, 1, 4, 5))
375 index_2_ms = Tensor(index_2, mstype.int32)
396 def construct(self, index_0, index_1, index_2): argument
412 index_2 = np.random.randint(3, size=(2, 1, 4, 5))
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sort_cpu_kernel.cc | 75 … comparator = [&input](size_t index_1, size_t index_2) { return input[index_1] > input[index_2]; }; in Launch() argument
77 … comparator = [&input](size_t index_1, size_t index_2) { return input[index_1] < input[index_2]; }; in Launch() argument
|
| D | topk_cpu_kernel.cc | 55 const std::function<bool(size_t, size_t)> comparator = [input](size_t index_1, size_t index_2) { in LaunchKernel() argument
56 return input[index_1] > input[index_2]; in LaunchKernel()
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_tensor_slice.py | 174 def construct(self, x, index_0, index_1, index_2): argument
175 ret = x[index_0, index_1, index_2] + self.const
224 def construct(self, tensor, index_0, index_1, index_2): argument
225 ret = tensor[0:2, index_0, index_1, 2, index_2, 0:3, ...] + self.const
234 def construct(self, tensor, index_0, index_1, index_2): argument
235 ret = tensor[0:2, index_0, index_1, ..., index_2, 2] + self.const
244 def construct(self, tensor, index_0, index_1, index_2): argument
245 ret = tensor[..., index_0, index_1, index_2, 3] + self.const
258 def construct(self, index_0, index_1, index_2): argument
259 self.param[0:2, index_0, index_1, 2, index_2, 0:3, ...] = self.value
[all …]
|
| /third_party/protobuf/src/google/protobuf/util/ |
| D | field_comparator.cc | 60 const FieldDescriptor* field, int index_1, int index_2, in Compare() argument
70 reflection_2->GetRepeated##METHOD(message_2, field, index_2))); \ in Compare()
101 message_2, field, index_2, &scratch2))); in Compare()
|
| D | field_comparator.h | 90 int index_2,
118 int index_2,
|
| D | message_differencer_unittest.cc | 2276 int index_2, in Compare() argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/ |
| D | strassen.cl | 64 int index_2 = (gidx + offset.z) * origin_shape + gidy + offset.w;
66 FLT4 result2 = input[index_2];
|
| /third_party/typescript/lib/ |
| D | typingsInstaller.js | 51136 var index_2 = pattern.elements.indexOf(declaration);
51142 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
51146 var indexType = getLiteralType(index_2);
|
| D | tsc.js | 42082 var index_2 = pattern.elements.indexOf(declaration);
42085 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
42089 var indexType = getLiteralType(index_2);
|
| D | tsserver.js | 51147 var index_2 = pattern.elements.indexOf(declaration);
51153 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
51157 var indexType = getLiteralType(index_2);
|
| D | typescriptServices.js | 51341 var index_2 = pattern.elements.indexOf(declaration);
51347 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
51351 var indexType = getLiteralType(index_2);
|
| D | tsserverlibrary.js | 51341 var index_2 = pattern.elements.indexOf(declaration);
51347 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
51351 var indexType = getLiteralType(index_2);
|
| D | typescript.js | 51341 var index_2 = pattern.elements.indexOf(declaration);
51347 mapType(parentType, function (t) { return sliceTupleType(t, index_2); }) :
51351 var indexType = getLiteralType(index_2);
|