Home
last modified time | relevance | path

Searched refs:kAllocatorAlignment (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dscoped_allocator_mgr_test.cc69 (Allocator::kAllocatorAlignment - in AlignmentPadding()
70 (521 * sizeof(float)) % Allocator::kAllocatorAlignment) % in AlignmentPadding()
71 Allocator::kAllocatorAlignment; in AlignmentPadding()
Dscoped_allocator_mgr.cc171 size_t overshoot = offset % Allocator::kAllocatorAlignment; in PopulateFields()
173 offset += (Allocator::kAllocatorAlignment - overshoot); in PopulateFields()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_constants.cc31 tensorflow::Allocator::kAllocatorAlignment;
/external/tensorflow/tensorflow/core/framework/
Dallocator.h84 static constexpr size_t kAllocatorAlignment = 64;
132 void* p = AllocateRaw(kAllocatorAlignment, sizeof(T) * num_elements, in Allocate()
Dallocator.cc40 constexpr size_t Allocator::kAllocatorAlignment; member in tensorflow::Allocator
/external/tensorflow/tensorflow/core/util/
Dmemmapped_file_system_writer.cc49 TF_RETURN_IF_ERROR(AdjustAlignment(Allocator::kAllocatorAlignment)); in SaveTensor()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime_test.cc32 EXPECT_EQ(cpu_function_runtime::kAlign, Allocator::kAllocatorAlignment); in TEST()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc132 bias_size * sizeof(float), Allocator::kAllocatorAlignment)); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc274 void* ptr = allocator->AllocateRaw(Allocator::kAllocatorAlignment, in TEST_F()
Dgpu_device.cc350 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc179 data = wrapped_->AllocateRaw(Allocator::kAllocatorAlignment, size, attrs); in Allocate()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer.cc297 CHECK_EQ(0, Allocator::kAllocatorAlignment % DataTypeSize(*dtype)) in AnalyzeInputs()