Searched refs:EmitBufferPointer (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 403 llvm::Value* data_address = EmitBufferPointer(data_slice, data_shape); in HandleInfeed()
406 llvm::Value* token_address = EmitBufferPointer( in HandleInfeed()
430 EmitBufferPointer(buffer, tuple_element_shape); in HandleInfeed()
570 EmitBufferPointer(destination_buffer, operand->shape()); in HandleSort()
1091 operand_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduceSingleReplica()
1152 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduceMultipleReplica()
1161 input_buffer_ptrs.push_back(EmitBufferPointer(input_slice, shape)); in HandleAllReduceMultipleReplica()
1162 output_buffer_ptrs.push_back(EmitBufferPointer(output_slice, shape)); in HandleAllReduceMultipleReplica()
1231 output_buffer_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllToAll()
1275 llvm::Value* input_buffer = EmitBufferPointer(input_slice, shape); in HandleCollectivePermute()
[all …]
|
| D | ir_emitter.h | 265 llvm::Value* EmitBufferPointer(const BufferAllocation::Slice& slice,
|