Home
last modified time | relevance | path

Searched refs:EIGEN_MAX_ALIGN_BYTES (Results 1 – 25 of 37) sorted by relevance

12

/external/eigen/Eigen/src/Core/util/
DConfigureVectorization.h136 #if defined(EIGEN_MAX_ALIGN_BYTES) && EIGEN_MAX_ALIGN_BYTES<EIGEN_MAX_STATIC_ALIGN_BYTES
138 #define EIGEN_MAX_STATIC_ALIGN_BYTES EIGEN_MAX_ALIGN_BYTES
165 #if defined(EIGEN_DONT_ALIGN) && defined(EIGEN_MAX_ALIGN_BYTES) && EIGEN_MAX_ALIGN_BYTES>0
166 #error EIGEN_MAX_ALIGN_BYTES and EIGEN_DONT_ALIGN are both defined with EIGEN_MAX_ALIGN_BYTES!=0. U…
170 #ifdef EIGEN_MAX_ALIGN_BYTES
171 #undef EIGEN_MAX_ALIGN_BYTES
173 #define EIGEN_MAX_ALIGN_BYTES 0 macro
174 #elif !defined(EIGEN_MAX_ALIGN_BYTES)
175 #define EIGEN_MAX_ALIGN_BYTES EIGEN_IDEAL_MAX_ALIGN_BYTES macro
178 #if EIGEN_IDEAL_MAX_ALIGN_BYTES > EIGEN_MAX_ALIGN_BYTES
[all …]
DConstants.h241 #if EIGEN_MAX_ALIGN_BYTES==128
243 #elif EIGEN_MAX_ALIGN_BYTES==64
245 #elif EIGEN_MAX_ALIGN_BYTES==32
247 #elif EIGEN_MAX_ALIGN_BYTES==16
249 #elif EIGEN_MAX_ALIGN_BYTES==8
251 #elif EIGEN_MAX_ALIGN_BYTES==0
254 #error Invalid value for EIGEN_MAX_ALIGN_BYTES
DMemory.h795 #if EIGEN_MAX_ALIGN_BYTES!=0 && !defined(EIGEN_HIP_DEVICE_COMPILE)
845 … (((EIGEN_MAX_ALIGN_BYTES>=16) && ((sizeof(Scalar)*(Size))%(EIGEN_MAX_ALIGN_BYTES )==0)) || \
846 … ((EIGEN_MAX_ALIGN_BYTES>=32) && ((sizeof(Scalar)*(Size))%(EIGEN_MAX_ALIGN_BYTES/2)==0)) || \
847 … ((EIGEN_MAX_ALIGN_BYTES>=64) && ((sizeof(Scalar)*(Size))%(EIGEN_MAX_ALIGN_BYTES/4)==0)) )))
/external/tensorflow/tensorflow/core/framework/
Dops_util.h52 #if EIGEN_MAX_ALIGN_BYTES == 0 in IsInnerDimsSizeAligned()
56 return bytes_per_dim0 % EIGEN_MAX_ALIGN_BYTES == 0; in IsInnerDimsSizeAligned()
68 #if EIGEN_MAX_ALIGN_BYTES == 0 in IsDim0SliceAligned()
71 bool start_aligned = (start * sizeof(T)) % EIGEN_MAX_ALIGN_BYTES == 0; in IsDim0SliceAligned()
76 bool end_aligned = (end_or_size * sizeof(T)) % EIGEN_MAX_ALIGN_BYTES == 0; in IsDim0SliceAligned()
Dtensor.h313 #if EIGEN_MAX_ALIGN_BYTES == 0 in IsAligned()
318 (reinterpret_cast<intptr_t>(ptr) % EIGEN_MAX_ALIGN_BYTES == 0); in IsAligned()
1063 EIGEN_MAX_ALIGN_BYTES)); in Tensor()
/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.cc61 if (EIGEN_MAX_ALIGN_BYTES == 0) return base_chunk_elts; in AlignedChunkElts()
62 if (EIGEN_MAX_ALIGN_BYTES <= elt_bytes) { in AlignedChunkElts()
64 DCHECK_EQ(0, elt_bytes % EIGEN_MAX_ALIGN_BYTES); in AlignedChunkElts()
69 DCHECK_EQ(0, EIGEN_MAX_ALIGN_BYTES % elt_bytes) in AlignedChunkElts()
71 << " EIGEN_MAX_ALIGN_BYTES=" << EIGEN_MAX_ALIGN_BYTES in AlignedChunkElts()
76 (chunk_bytes < EIGEN_MAX_ALIGN_BYTES) in AlignedChunkElts()
77 ? (EIGEN_MAX_ALIGN_BYTES - chunk_bytes) in AlignedChunkElts()
78 : (EIGEN_MAX_ALIGN_BYTES - (chunk_bytes % EIGEN_MAX_ALIGN_BYTES)); in AlignedChunkElts()
81 DCHECK_EQ(0, ((base_chunk_elts * elt_bytes) % EIGEN_MAX_ALIGN_BYTES)) in AlignedChunkElts()
83 << " EIGEN_MAX_ALIGN_BYTES=" << EIGEN_MAX_ALIGN_BYTES in AlignedChunkElts()
/external/tensorflow/tensorflow/core/kernels/
Dops_util_test.cc320 #if EIGEN_MAX_ALIGN_BYTES == 0 in TEST_F()
328 Tensor t(DT_FLOAT, TensorShape({EIGEN_MAX_ALIGN_BYTES * 2})); in TEST_F()
330 int64_t end = EIGEN_MAX_ALIGN_BYTES; in TEST_F()
340 #if EIGEN_MAX_ALIGN_BYTES > 0
342 Tensor t(DT_FLOAT, TensorShape({EIGEN_MAX_ALIGN_BYTES * 2})); in TEST_F()
344 int64_t end = EIGEN_MAX_ALIGN_BYTES + 1; in TEST_F()
355 #if EIGEN_MAX_ALIGN_BYTES == 0 in TEST_F()
365 int64_t inner_dim_size = EIGEN_MAX_ALIGN_BYTES; in TEST_F()
378 #if EIGEN_MAX_ALIGN_BYTES > 0
381 int64_t inner_dim_size = EIGEN_MAX_ALIGN_BYTES + 1; in TEST_F()
Dwhile_op_test.cc123 #if EIGEN_MAX_ALIGN_BYTES == 0 in TEST_F()
126 return tensorflow::port::AlignedMalloc(size, EIGEN_MAX_ALIGN_BYTES); in TEST_F()
136 #if EIGEN_MAX_ALIGN_BYTES == 0 in TEST_F()
139 mem->opaque = tensorflow::port::AlignedMalloc(size, EIGEN_MAX_ALIGN_BYTES); in TEST_F()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_util.cc39 if (EIGEN_MAX_ALIGN_BYTES != 0 && in ShouldReuseTensorMemory()
40 reinterpret_cast<intptr_t>(tensor->data.raw) % EIGEN_MAX_ALIGN_BYTES) { in ShouldReuseTensorMemory()
76 return tensorflow::cpu_allocator()->AllocateRaw(EIGEN_MAX_ALIGN_BYTES, in MaybeAllocateTensorflowBuffer()
/external/eigen/test/
Ddynalloc.cpp12 #if EIGEN_MAX_ALIGN_BYTES>0
13 #define ALIGNMENT EIGEN_MAX_ALIGN_BYTES
110 #if EIGEN_MAX_ALIGN_BYTES>0 && (!EIGEN_HAS_CXX17_OVERALIGN) in check_custom_new_delete()
Dmapped_matrix.cpp27 …Scalar* array3unaligned = (internal::UIntPtr(array3)%EIGEN_MAX_ALIGN_BYTES) == 0 ? array3+1 : arra… in map_class_vector()
67 …Scalar* array3unaligned = (internal::UIntPtr(array3)%EIGEN_MAX_ALIGN_BYTES) == 0 ? array3+1 : arra… in map_class_matrix()
130 …Scalar* array3unaligned = internal::UIntPtr(array3)%EIGEN_MAX_ALIGN_BYTES == 0 ? array3+1 : array3; in map_static_methods()
Dmapstride.cpp71 …= (Scalar*)(((internal::UIntPtr(a_array2)+EIGEN_MAX_ALIGN_BYTES-1)/EIGEN_MAX_ALIGN_BYTES)*EIGEN_MA… in map_class_matrix()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
Dtf_jitrt_executor.cc83 copts.alignment = EIGEN_MAX_ALIGN_BYTES; in Compile()
152 #if EIGEN_MAX_ALIGN_BYTES == 0 in IsAligned()
155 return reinterpret_cast<intptr_t>(ptr) % EIGEN_MAX_ALIGN_BYTES == 0; in IsAligned()
/external/eigen/doc/
DPreprocessorDirectives.dox104 …- \b \c EIGEN_MAX_ALIGN_BYTES - Must be a power of two, or 0. Defines an upper bound on the memory…
105 …EIGEN_MAX_ALIGN_BYTES=16. In the other way round, since by default AVX implies 32 bytes alignment …
106 …EIGEN_MAX_ALIGN_BYTES but for statically allocated data only. By default, if only \c EIGEN_MAX_AL…
134 …- \c EIGEN_DONT_ALIGN - Deprecated, it is a synonym for \c EIGEN_MAX_ALIGN_BYTES=0. It disables al…
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/
Dtf_jitrt_benchmark_test.cc73 copts.alignment = EIGEN_MAX_ALIGN_BYTES; in BM_InstantiateExecutable()
/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dtensor_util_test.cc48 host_context->AllocateBytes(size, /*alignment=*/EIGEN_MAX_ALIGN_BYTES); in TEST()
/external/tensorflow/third_party/eigen3/
Deigen_archive.BUILD52 "EIGEN_MAX_ALIGN_BYTES=64",
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc40 void* data = allocator->AllocateRaw(EIGEN_MAX_ALIGN_BYTES, len); in allocate_tensor()
108 reinterpret_cast<intptr_t>(data) % std::max(1, EIGEN_MAX_ALIGN_BYTES) != in TF_NewTensor()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceSycl.h164 #if EIGEN_MAX_ALIGN_BYTES > 0 in allocate()
165 size_t align = num_bytes % EIGEN_MAX_ALIGN_BYTES; in allocate()
167 num_bytes += EIGEN_MAX_ALIGN_BYTES - align; in allocate()
175 #if EIGEN_MAX_ALIGN_BYTES > 0 in allocate_temp()
176 size_t align = num_bytes % EIGEN_MAX_ALIGN_BYTES; in allocate_temp()
178 num_bytes += EIGEN_MAX_ALIGN_BYTES - align; in allocate_temp()
DTensorTraits.h32 #if EIGEN_MAX_ALIGN_BYTES>0
/external/eigen/Eigen/src/Core/
DGeneralProduct.h188 …<Scalar,EIGEN_SIZE_MIN_PREFER_FIXED(Size,MaxSize)+(ForceAlignment?EIGEN_MAX_ALIGN_BYTES:0),0> m_da…
191 …>((internal::UIntPtr(m_data.array) & ~(std::size_t(EIGEN_MAX_ALIGN_BYTES-1))) + EIGEN_MAX_ALIGN_BY…
DMapBase.h194 #if EIGEN_MAX_ALIGN_BYTES>0 in EIGEN_DEFAULT_COPY_CONSTRUCTOR()
/external/tensorflow/tensorflow/lite/kernels/
Deigen_support.cc45 kDefaultTensorAlignment % EIGEN_MAX_ALIGN_BYTES == 0,
/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
Dbenchmark.cc61 copts.alignment = EIGEN_MAX_ALIGN_BYTES; in CreateJitExecutable()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_kernels.cc159 opts.alignment = EIGEN_MAX_ALIGN_BYTES; // Eigen included by tensor.h in CompileImpl()

12