Searched refs:output_buffer_ptrs (Results 1 – 1 of 1) sorted by relevance
1215 std::vector<llvm::Value*> output_buffer_ptrs; in HandleAllReduceMultipleReplica() local1226 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduceMultipleReplica()1236 output_buffer_ptrs.push_back(EmitBufferPointer(output_slice, shape)); in HandleAllReduceMultipleReplica()1242 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllReduceMultipleReplica()1301 std::vector<llvm::Value*> output_buffer_ptrs; in HandleAllToAll() local1310 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllToAll()1319 EncodeArrayFunctionArguments(output_buffer_ptrs, "output_buffers", &b_); in HandleAllToAll()1338 llvm_ir::EmitTuple(GetIrArrayFor(instruction), output_buffer_ptrs, &b_); in HandleAllToAll()