Searched refs:sliced_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/extension/aten_util/test/ |
| H A D | aten_bridge_test.cpp | 84 auto sliced_tensor = at_tensor.slice(1, 0, 2); in TEST() local
85 auto sliced_tensor_contig = sliced_tensor.contiguous(); in TEST()
87 sliced_tensor.sizes().begin(), sliced_tensor.sizes().end()); in TEST()
92 auto dtype = torchToExecuTorchScalarType(sliced_tensor.options().dtype()); in TEST()
96 sliced_tensor.dim(), in TEST()
104 EXPECT_NE(sliced_tensor.const_data_ptr(), etensor.const_data_ptr()); in TEST()
109 auto sliced_tensor = at_tensor.slice(1, 0, 2); in TEST() local
110 auto sliced_tensor_contig = sliced_tensor.contiguous(); in TEST()
112 sliced_tensor.sizes().begin(), sliced_tensor.sizes().end()); in TEST()
116 auto dtype = torchToExecuTorchScalarType(sliced_tensor.options().dtype()); in TEST()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 249 auto sliced_tensor = wei.chunk(4, 0); in permute_wei_for_miopen() local
250 … permuted_wei = at::cat({sliced_tensor[0], sliced_tensor[1], sliced_tensor[3], sliced_tensor[2]}); in permute_wei_for_miopen()
253 auto sliced_tensor = wei.chunk(3, 0); in permute_wei_for_miopen() local
254 permuted_wei = at::cat({sliced_tensor[1], sliced_tensor[0], sliced_tensor[2]}); in permute_wei_for_miopen()
|
| /aosp_15_r20/external/executorch/backends/cadence/aot/ |
| H A D | reorder_ops.py | 278 sliced_tensor = inp.args[0]
279 assert isinstance(sliced_tensor, torch.fx.Node)
280 slice_input_shape = get_shape(self.graph_module, sliced_tensor)
|