Searched refs:GetAllocatedBuffer (Results 1 – 2 of 2) sorted by relevance
817 U* weights_data = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute()1211 U* weights_data_tf = reinterpret_cast<U*>(weights.GetAllocatedBuffer()); in Compute()1266 U* diff_weights_data = static_cast<U*>(diff_weights.GetAllocatedBuffer()); in Compute()
1409 inline void* GetAllocatedBuffer() { return allocated_buffer_; } in GetAllocatedBuffer() function