Searched defs:TensorView (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/ |
| H A D | predicated_tile_access_iterator_residual_last.h | 118 using TensorView = TensorView<Element, Layout>; variable
477 using TensorView = TensorView<Element, Layout>; variable
702 using TensorView = TensorView<Element, Layout>; variable
924 using TensorView = TensorView<Element, Layout>; variable
1258 using TensorView = TensorView<Element, Layout>; variable
1477 using TensorView = TensorView<Element, Layout>; variable
1699 using TensorView = TensorView<Element, Layout>; variable
1926 using TensorView = TensorView<Element, Layout>; variable
|
| H A D | predicated_tile_iterator_residual_last.h | 193 using TensorView = TensorView<Element, Layout>; variable
489 using TensorView = TensorView<Element, Layout>; variable
721 using TensorView = TensorView<Element, Layout>; variable
950 using TensorView = TensorView<Element, Layout>; variable
1241 using TensorView = TensorView<Element, Layout>; variable
1468 using TensorView = TensorView<Element, Layout>; variable
1698 using TensorView = TensorView<Element, Layout>; variable
1927 using TensorView = TensorView<Element, Layout>; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | README.md | 35 ### TensorView subsection
|
| H A D | tensor_view.h | 206 TensorView(const absl::Span<int> shape, void *data, in TensorView() function
|
| /aosp_15_r20/external/libtextclassifier/native/utils/ |
| H A D | tensor-view.h | 35 TensorView(const T* data, const std::vector<int>& shape) in TensorView() function
|
| /aosp_15_r20/external/angle/third_party/spirv-headers/src/include/spirv/unified1/ |
| H A D | spv.d | 1484 TensorView = 0x00000001, enumerator
|
| H A D | spirv.cs | 1473 TensorView = 0, enumerator
1481 TensorView = 0x00000001, enumerator
|
| H A D | spirv.json | 1405 "TensorView": 0, number
|