Searched refs:kAllocatorAlignment (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr_test.cc | 69 (Allocator::kAllocatorAlignment - in AlignmentPadding()
70 (521 * sizeof(float)) % Allocator::kAllocatorAlignment) % in AlignmentPadding()
71 Allocator::kAllocatorAlignment; in AlignmentPadding()
|
| D | scoped_allocator_mgr.cc | 183 size_t overshoot = offset % Allocator::kAllocatorAlignment; in PopulateFields()
185 size_t alignment_bytes = Allocator::kAllocatorAlignment - overshoot; in PopulateFields()
|
| D | scoped_allocator.cc | 114 << Allocator::kAllocatorAlignment << " and hence we allocated " in AllocateRaw()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_constants.cc | 31 tensorflow::Allocator::kAllocatorAlignment;
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | cpu_function_runtime_test.cc | 31 EXPECT_EQ(xla::cpu_function_runtime::kAlign, Allocator::kAllocatorAlignment); in TEST()
33 Allocator::kAllocatorAlignment); in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.cc | 46 data = wrapped_->AllocateRaw(tensorflow::Allocator::kAllocatorAlignment, in Allocate()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator.cc | 52 constexpr size_t Allocator::kAllocatorAlignment; member in tensorflow::Allocator
|
| D | typed_allocator.h | 47 raw_allocator->AllocateRaw(Allocator::kAllocatorAlignment, in Allocate()
|
| D | allocator.h | 184 static constexpr size_t kAllocatorAlignment = 64;
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 68 ->AllocateRaw(Allocator::kAllocatorAlignment, in _mlir_ciface_tf_alloc()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | memmapped_file_system_writer.cc | 49 TF_RETURN_IF_ERROR(AdjustAlignment(Allocator::kAllocatorAlignment)); in SaveTensor()
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 132 bias_size * sizeof(float), Allocator::kAllocatorAlignment)); in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 58 static_cast<int64>(tensorflow::Allocator::kAllocatorAlignment))), in RedzoneAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 153 alloc->AllocateRaw(Allocator::kAllocatorAlignment, total_bytes)); in SetProtoFromGPU()
|
| D | gpu_device_test.cc | 401 void* ptr = allocator->AllocateRaw(Allocator::kAllocatorAlignment, in TEST_F()
|
| D | gpu_device.cc | 410 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer.cc | 467 CHECK_EQ(0, Allocator::kAllocatorAlignment % DataTypeSize(*dtype)) in AnalyzeInputs()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 736 tensorflow::Allocator::kAllocatorAlignment, size); in BufferFromHostBuffer()
|