Home
last modified time | relevance | path

Searched refs:pointer_size_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgeneric_transfer_manager.cc36 : platform_id_(platform_id), pointer_size_(pointer_size) {} in GenericTransferManager()
160 return ShapeUtil::ByteSizeOf(shape, pointer_size_); in GetByteSizeRequirement()
Dgeneric_transfer_manager.h71 const size_t pointer_size_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.h114 return [pointer_size = pointer_size_](const Shape& shape) { in ShapeSizeBytesFunction()
144 const int64 pointer_size_; variable
Dgpu_compiler.cc145 pointer_size_(llvm::DataLayout(data_layout) in GpuCompiler()
525 GpuHloSchedule::Build(*hlo_module, *stream_assignment, pointer_size_)); in RunHloPassesAndBufferAssignement()
529 return GpuCompiler::GetSizeOfShape(buffer_value.shape(), pointer_size_); in RunHloPassesAndBufferAssignement()
907 GetCanShareBuffer(), pointer_size_, profile_index_map.get(), &llvm_module, in RunBackend()