Searched refs:MakeOnStackBuffer (Results 1 – 3 of 3) sorted by relevance
44 return BufferInfo::MakeOnStackBuffer(on_stack_buffer_size); in SizesToBufferInfos()161 CheckRoundTripIsOk(BufferInfo::MakeOnStackBuffer(0)); in TEST()162 CheckRoundTripIsOk(BufferInfo::MakeOnStackBuffer(4)); in TEST()
29 buffer_infos.push_back(BufferInfo::MakeOnStackBuffer(allocation.size())); in CreateBufferInfosFromBufferAssignment()
101 static BufferInfo MakeOnStackBuffer(tensorflow::uint64 size) { in MakeOnStackBuffer() function