Searched refs:at_aligned (Results 1 – 1 of 1) sorted by relevance
638 tflite::gpu::Tensor<Linear, DataType::FLOAT32> at_aligned; in UploadAt() local639 at_aligned.shape = Linear(4 * 8); in UploadAt()640 at_aligned.data.resize(4 * 8); in UploadAt()644 at_aligned.data[y * 8 + x] = at_mat[y * 6 + x]; in UploadAt()646 at_aligned.data[y * 8 + 6] = 0.0f; in UploadAt()647 at_aligned.data[y * 8 + 7] = 0.0f; in UploadAt()653 desc.UploadLinearData(at_aligned); in UploadAt()