Searched refs:operand_buffer (Results 1 – 4 of 4) sorted by relevance
61 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine() local62 const float* const operand_data = operand_buffer.data.data(); in FuseAddOrSubParamsIntoPrecedingAffine()133 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine() local134 const float* const operand_data = operand_buffer.data.data(); in FuseMulOrDivParamsIntoPrecedingAffine()
257 const HloBuffer* operand_buffer = in RunComputation() local259 if (operand_buffer->values().size() > 1) { in RunComputation()
1239 TF_ASSIGN_OR_RETURN(auto operand_buffer, in EmitCholeskyThunk()1250 if (operand_buffer != a_buffer) { in EmitCholeskyThunk()1253 /*source_address=*/operand_buffer, in EmitCholeskyThunk()1843 auto operand_buffer = *maybe_slice; in EmitFusion() local1845 if (operand_buffer != destination_buffer) { in EmitFusion()1848 /*source_address=*/operand_buffer, in EmitFusion()2214 TF_ASSIGN_OR_RETURN(auto operand_buffer, in EmitScatter()2220 if (operand_buffer != output_buffer) { in EmitScatter()2223 /*source_address=*/operand_buffer, in EmitScatter()
1318 se::DeviceMemoryBase operand_buffer = GetDeviceAddress(operand); in operator ()() local1328 stream->ThenMemcpy(&a_buffer, operand_buffer, operand_buffer.size()); in operator ()()