Searched refs:ShardedVectorType (Results 1 – 2 of 2) sorted by relevance
371 using ShardedVectorType = std::vector<llvm::Type*>; variable375 ShardedVectorType CreateShardedVectorType(PrimitiveType element_type,400 const ShardedVectorType& accumulator_type, HloInstruction* init_value,
1477 IrEmitter::ShardedVectorType IrEmitter::CreateShardedVectorType( in CreateShardedVectorType()1484 ShardedVectorType sharded_vector_type; in CreateShardedVectorType()1533 const ShardedVectorType& accumulator_type, HloInstruction* init_value, in EmitInnerLoopForVectorizedReduction()1737 ShardedVectorType vector_type = CreateShardedVectorType( in EmitVectorizedReduce()1770 ShardedVectorType vector_type = CreateShardedVectorType( in EmitVectorizedReduce()