| /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 | 117 using TensorRef = TensorRef<Element, Layout>; variable
476 using TensorRef = TensorRef<Element, Layout>; variable
701 using TensorRef = TensorRef<Element, Layout>; variable
923 using TensorRef = TensorRef<Element, Layout>; variable
1257 using TensorRef = TensorRef<Element, Layout>; variable
1476 using TensorRef = TensorRef<Element, Layout>; variable
1698 using TensorRef = TensorRef<Element, Layout>; variable
1925 using TensorRef = TensorRef<Element, Layout>; variable
|
| H A D | predicated_tile_iterator_residual_last.h | 192 using TensorRef = TensorRef<Element, Layout>; variable
488 using TensorRef = TensorRef<Element, Layout>; variable
720 using TensorRef = TensorRef<Element, Layout>; variable
949 using TensorRef = TensorRef<Element, Layout>; variable
1240 using TensorRef = TensorRef<Element, Layout>; variable
1467 using TensorRef = TensorRef<Element, Layout>; variable
1697 using TensorRef = TensorRef<Element, Layout>; variable
1926 using TensorRef = TensorRef<Element, Layout>; variable
|
| H A D | warp_iterator_from_smem.h | 99 using TensorRef = TensorRef<Element, Layout>; variable
|
| H A D | epilogue_predicated_tile_iterator.h | 83 using TensorRef = TensorRef<Element, Layout>; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/ |
| H A D | custom_call.h | 76 struct TensorRef { struct
77 llvm::ArrayRef<int64_t> shape;
78 llvm::ArrayRef<T> data;
|
| H A D | custom_call.cc | 31 using TensorRef = CustomCall::TensorRef<T>; typedef
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/warp/ |
| H A D | mma_tensorop_dequantizer.h | 134 using TensorRef = TensorRef<ElementScale, Layout>; variable
242 using TensorRef = TensorRef<ElementScale, Layout>; variable
336 using TensorRef = TensorRef<ElementScale, Layout>; variable
422 using TensorRef = TensorRef<ElementScale, Layout>; variable
|
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorRef.h | 155 EIGEN_STRONG_INLINE TensorRef() : m_evaluator(NULL) { in TensorRef() function
159 …EIGEN_STRONG_INLINE TensorRef(const Expression& expr) : m_evaluator(new internal::TensorLazyEvalua… in TensorRef() function
175 TensorRef(const TensorRef& other) : m_evaluator(other.m_evaluator) { in TensorRef() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/templates/ |
| H A D | TensorBody.h | 68 class TensorRef; variable
100 friend TensorRef; variable
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/containers/ |
| H A D | Constant.cpp | 13 TensorRef::TensorRef( in TensorRef() function in vkcompute::TensorRef
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | Tensor.h | 60 TensorRef(const TensorBase& src) in TensorRef() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/ |
| H A D | custom_mma_base.h | 106 using TensorRef = TensorRef<Element, OperandLayout>; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/epilogue/ |
| H A D | epilogue_pipelined.h | 166 using TensorRef = typename OutputTileIterator::TensorRef; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 353 AsyncValueRef<Tensor> TensorInterface::TensorRef() const { in TensorRef() function in tfrt::tf::TensorInterface
|