Searched refs:ShardedVectorType (Results 1 – 2 of 2) sorted by relevance
370 using ShardedVectorType = std::vector<llvm::Type*>; variable374 ShardedVectorType CreateShardedVectorType(PrimitiveType element_type,399 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
1512 IrEmitter::ShardedVectorType IrEmitter::CreateShardedVectorType( in CreateShardedVectorType()1519 ShardedVectorType sharded_vector_type; in CreateShardedVectorType()1568 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction()1757 ShardedVectorType vector_type = CreateShardedVectorType( in EmitVectorizedReduce()1790 ShardedVectorType vector_type = CreateShardedVectorType( in EmitVectorizedReduce()