Searched refs:output_buffer_ptrs (Results 1 – 1 of 1) sorted by relevance
1141 std::vector<llvm::Value*> output_buffer_ptrs; in HandleAllReduceMultipleReplica() local1152 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduceMultipleReplica()1162 output_buffer_ptrs.push_back(EmitBufferPointer(output_slice, shape)); in HandleAllReduceMultipleReplica()1168 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllReduceMultipleReplica()1222 std::vector<llvm::Value*> output_buffer_ptrs; in HandleAllToAll() local1231 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllToAll()1240 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllToAll()1259 llvm_ir::EmitTuple(GetIrArrayFor(instruction), output_buffer_ptrs, &b_); in HandleAllToAll()