Searched refs:elem_prim_ty (Results 1 – 1 of 1) sorted by relevance
1376 PrimitiveType elem_prim_ty = hlo->shape().element_type(); in ConvertValueForDistribution() local1378 llvm_ir::PrimitiveTypeToIrType(elem_prim_ty, module_); in ConvertValueForDistribution()1470 EmitErfcInv(elem_prim_ty, FMul(llvm::ConstantFP::get(elem_ir_ty, 2.0), in ConvertValueForDistribution()1486 int32 GetNumberOfElementsPerPhiloxRngSample(PrimitiveType elem_prim_ty) { in GetNumberOfElementsPerPhiloxRngSample() argument1489 switch (elem_prim_ty) { in GetNumberOfElementsPerPhiloxRngSample()1503 LOG(FATAL) << "Unrecognized primitive type for RNG " << elem_prim_ty; in GetNumberOfElementsPerPhiloxRngSample()