Home
last modified time | relevance | path

Searched refs:index_0 (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/tests/st/pynative/
Dtest_tensor_getitem.py167 def construct(self, x, index_0, index_1, index_2): argument
168 ret0 = x[index_0] + self.const0
169 ret1 = x[index_0, index_1] + self.const1
170 ret2 = x[index_0, index_1, index_2] + self.const2
183 index_0 = np.random.randint(6, size=(3, 4, 5)).astype(np.int32)
187 index_0_ms = Tensor(index_0)
191 assert np.all(output0.asnumpy() == input_x[index_0] + np.ones([4, 5, 8, 10]))
192 assert np.all(output1.asnumpy() == input_x[index_0, index_1] + np.ones([3, 4, 5, 10]))
193 assert np.all(output2.asnumpy() == input_x[index_0, index_1, index_2] + np.ones([5, 3, 4, 5]))
206 def construct(self, tensor, index_0, index_1): argument
[all …]
/third_party/mindspore/tests/ut/python/ops/
Dtest_tuple_slice.py34 self.index_0 = Tensor(3)
40 tensor_tuple_slice1 = tensor_tuple[:self.index_0]
58 self.index_0 = Tensor(-6)
64 tensor_tuple_slice3 = tensor_tuple[self.index_0:3]
101 self.index_0 = Tensor([2, 3])
104 return tensor_tuple[1:self.index_0]
112 self.index_0 = Tensor([2.1])
115 return tensor_tuple[1:self.index_0]
Dtest_tensor_slice.py164 def construct(self, x, index_0, index_1): argument
165 ret = x[index_0, index_1] + self.const
174 def construct(self, x, index_0, index_1, index_2): argument
175 ret = x[index_0, index_1, index_2] + self.const
184 def construct(self, tensor, index_0, index_1): argument
185 ret = tensor[index_0, index_1, 0:3, ..., 0:5, 3] + self.const
194 def construct(self, tensor, index_0, index_1): argument
195 ret = tensor[0:3, index_0, ..., index_1, 3, 0:5] + self.const
204 def construct(self, tensor, index_0, index_1): argument
205 ret = tensor[0, index_0, index_1, ..., 3] + self.const
[all …]
/third_party/mesa3d/src/intel/compiler/
Dbrw_vec4_generator.cpp1105 struct brw_reg index_0 = suboffset(vec1(index), 0); in generate_oword_dual_block_offsets() local
1112 brw_MOV(p, m1_0, index_0); in generate_oword_dual_block_offsets()
/third_party/vk-gl-cts/external/vulkancts/modules/vulkan/spirv_assembly/
DvktSpvAsmInstructionTests.cpp19084 int index_0 = rnd.getInt(0, column-1); in createMatrixCompositeCases() local
19098 params["indexes"] = numberToString(index_0) + " " + numberToString(index_1); in createMatrixCompositeCases()