Searched refs:GetAllocatedBuffer (Results 1 – 2 of 2) sorted by relevance
622 T* weights_data = reinterpret_cast<T*>(weights.GetAllocatedBuffer()); in Compute()947 T* weights_data_tf = reinterpret_cast<T*>(weights.GetAllocatedBuffer()); in Compute()1010 T* diff_weights_data = static_cast<T*>(diff_weights.GetAllocatedBuffer()); in Compute()
1826 inline void* GetAllocatedBuffer() { return allocated_buffer_; } in GetAllocatedBuffer() function