Home
last modified time | relevance | path

Searched refs:MakeOnStackBuffer (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dcpu_function_runtime_test.cc43 return BufferInfo::MakeOnStackBuffer(on_stack_buffer_size); in SizesToBufferInfos()
160 CheckRoundTripIsOk(BufferInfo::MakeOnStackBuffer(0)); in TEST()
161 CheckRoundTripIsOk(BufferInfo::MakeOnStackBuffer(4)); in TEST()
Dcpu_function_runtime.h99 static BufferInfo MakeOnStackBuffer(uint64 size) { in MakeOnStackBuffer() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc28 buffer_infos.push_back(BufferInfo::MakeOnStackBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()