Searched defs:TensorMap (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| H A D | TensorMap.h | 80 EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr) : m_data(dataPtr), m_dimensions() { in TensorMap() function
87 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index firstDimension, IndexTypes... othe… in TensorMap() function
93 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index firstDimension) : m_data(dataPtr),… in TensorMap() function
98 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2) : m_data(dataPtr… in TensorMap() function
102 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3) : m_… in TensorMap() function
106 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3, Inde… in TensorMap() function
110 …EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, Index dim1, Index dim2, Index dim3, Inde… in TensorMap() function
115 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, const array<Index, Num… in TensorMap() function
120 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorMap(StoragePointerType dataPtr, const Dimensions& dime… in TensorMap() function
124 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorMap(PlainObjectType& tensor) in TensorMap() function
|
| H A D | README.md | 581 ### TensorMap subsection
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_map.h | 69 TensorMap() : tensors_(new Tensors) {} in TensorMap() function
72 TensorMap(const TensorMap& other) : tensors_(other.tensors_) { in TensorMap() function
76 TensorMap(TensorMap&& rhs) : tensors_(rhs.tensors_) { in TensorMap() function
|
| H A D | conv_grad_ops_3d.cc | 535 TensorMap; in Compute() typedef
1048 TensorMap; in Compute() typedef
|
| H A D | conv_grad_filter_ops.cc | 564 TensorMap; in Compute() typedef
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/simple_aggregation/ |
| H A D | simple_aggregation_protocol.h | 166 using TensorMap = absl::flat_hash_map<std::string, Tensor>; variable
|