Searched refs:kAllocatorAlignment (Results 1 – 12 of 12) sorted by relevance
69 (Allocator::kAllocatorAlignment - in AlignmentPadding()70 (521 * sizeof(float)) % Allocator::kAllocatorAlignment) % in AlignmentPadding()71 Allocator::kAllocatorAlignment; in AlignmentPadding()
171 size_t overshoot = offset % Allocator::kAllocatorAlignment; in PopulateFields()173 offset += (Allocator::kAllocatorAlignment - overshoot); in PopulateFields()
31 tensorflow::Allocator::kAllocatorAlignment;
84 static constexpr size_t kAllocatorAlignment = 64;132 void* p = AllocateRaw(kAllocatorAlignment, sizeof(T) * num_elements, in Allocate()
40 constexpr size_t Allocator::kAllocatorAlignment; member in tensorflow::Allocator
49 TF_RETURN_IF_ERROR(AdjustAlignment(Allocator::kAllocatorAlignment)); in SaveTensor()
32 EXPECT_EQ(cpu_function_runtime::kAlign, Allocator::kAllocatorAlignment); in TEST()
132 bias_size * sizeof(float), Allocator::kAllocatorAlignment)); in Compute()
274 void* ptr = allocator->AllocateRaw(Allocator::kAllocatorAlignment, in TEST_F()
350 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()
179 data = wrapped_->AllocateRaw(Allocator::kAllocatorAlignment, size, attrs); in Allocate()
297 CHECK_EQ(0, Allocator::kAllocatorAlignment % DataTypeSize(*dtype)) in AnalyzeInputs()