Searched refs:operand_buffer (Results 1 – 3 of 3) 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()
254 const HloBuffer* operand_buffer = in RunComputation() local256 if (operand_buffer->values().size() > 1) { in RunComputation()
1497 TF_ASSIGN_OR_RETURN(auto operand_buffer, in EmitCholeskyThunkFromMlir()1508 if (operand_buffer != a_buffer) { in EmitCholeskyThunkFromMlir()1511 /*source_address=*/operand_buffer, in EmitCholeskyThunkFromMlir()2097 auto operand_buffer = *maybe_slice; in EmitCopyForMlir() local2099 if (operand_buffer != destination_buffer) { in EmitCopyForMlir()2102 /*source_address=*/operand_buffer, in EmitCopyForMlir()2504 TF_ASSIGN_OR_RETURN(auto operand_buffer, in EmitScatterFromMlir()2510 if (operand_buffer != output_buffer) { in EmitScatterFromMlir()2513 /*source_address=*/operand_buffer, in EmitScatterFromMlir()