Searched refs:input_buffer_ptrs (Results 1 – 1 of 1) sorted by relevance
1140 std::vector<llvm::Value*> input_buffer_ptrs; in HandleAllReduceMultipleReplica() local1153 input_buffer_ptrs.push_back(GetEmittedValueFor(op)); in HandleAllReduceMultipleReplica()1161 input_buffer_ptrs.push_back(EmitBufferPointer(input_slice, shape)); in HandleAllReduceMultipleReplica()1166 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllReduceMultipleReplica()1221 std::vector<llvm::Value*> input_buffer_ptrs; in HandleAllToAll() local1232 input_buffer_ptrs.push_back(GetEmittedValueFor(op)); in HandleAllToAll()1238 EncodeArrayFunctionArguments(input_buffer_ptrs, "input_buffers", &b_); in HandleAllToAll()