Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime.cc58 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n, in AlignedBufferBytes() function
77 AlignedBufferBytes(buffer_infos, n, allocate_entry_params); in MallocContiguousBuffers()
Dcpu_function_runtime_test.cc54 return AlignedBufferBytes(buffer_infos.data(), n, in AlignedBufferBytesFromSizes()
66 TEST(XlaCompiledCpuFunctionTest, AlignedBufferBytes) { in TEST() argument
Dcpu_function_runtime.h140 size_t AlignedBufferBytes(const BufferInfo* buffer_infos, size_t n,
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen.cc401 const size_t arg_bytes_aligned = cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()
405 const size_t temp_bytes_aligned = cpu_function_runtime::AlignedBufferBytes( in GenerateHeader()