Searched refs:operand_buffer (Results 1 – 3 of 3) sorted by relevance
213 for (const BufferValue* operand_buffer : used_buffers[instruction]) { in RunComputation() local214 if (IgnoreBuffer(operand_buffer)) { in RunComputation()218 << operand_buffer->ToString(); in RunComputation()219 auto it = live_buffers.find(operand_buffer); in RunComputation()224 operand_buffers_to_free.push_back(operand_buffer); in RunComputation()255 for (const BufferValue* operand_buffer : operand_buffers_to_free) { in RunComputation() local256 if (reused_buffers.contains(operand_buffer)) { in RunComputation()259 if (buffer->instruction()->IsUserOf(operand_buffer->instruction()) && in RunComputation()262 operand_buffer->instruction(), operand_buffer->index(), in RunComputation()265 << operand_buffer->ToString(); in RunComputation()[all …]
48 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine() local49 const float* const operand_data = operand_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine()119 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local120 const float* const operand_data = operand_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine()
497 auto operand_buffer = GetAllocationSlice(*custom_call->operand(0)); in HandleCustomCall() local506 if (operand_buffer != a_buffer) { in HandleCustomCall()508 /*source_address=*/operand_buffer, in HandleCustomCall()554 auto operand_buffer = GetAllocationSlice(*hlo->operand(1)); in HandleTriangularSolve() local556 if (operand_buffer != destination_buffer) { in HandleTriangularSolve()558 /*source_address=*/operand_buffer, in HandleTriangularSolve()1037 auto operand_buffer = GetAllocationSlice(*operand); in HandleScatter() local1039 if (operand_buffer != destination_buffer) { in HandleScatter()1041 /*source_address=*/operand_buffer, in HandleScatter()