Searched refs:input_buffer_ptrs (Results 1 – 1 of 1) sorted by relevance
1214 std::vector<llvm::Value*> input_buffer_ptrs; in HandleAllReduceMultipleReplica() local1227 input_buffer_ptrs.push_back(GetEmittedValueFor(op)); in HandleAllReduceMultipleReplica()1235 input_buffer_ptrs.push_back(EmitBufferPointer(input_slice, shape)); in HandleAllReduceMultipleReplica()1240 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllReduceMultipleReplica()1300 std::vector<llvm::Value*> input_buffer_ptrs; in HandleAllToAll() local1311 input_buffer_ptrs.push_back(GetEmittedValueFor(op)); in HandleAllToAll()1317 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllToAll()