Searched refs:EmitBufferPointer (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 339 llvm::Value* data_address = EmitBufferPointer(data_slice, data_shape); in HandleInfeed()
342 llvm::Value* token_address = EmitBufferPointer( in HandleInfeed()
366 EmitBufferPointer(buffer, tuple_element_shape); in HandleInfeed()
545 EmitBufferPointer(destination_buffer, operand->shape()); in HandleSort()
1340 operand_ptrs.push_back(EmitBufferPointer(out_slice, operand_shape)); in HandleAllReduce()
2283 llvm::Value* addr = EmitBufferPointer(slice, elem_shape); in HandleCustomCall()
2919 llvm::Value* IrEmitter::EmitBufferPointer(const BufferAllocation::Slice& slice, in EmitBufferPointer() function in xla::cpu::IrEmitter
2936 llvm::Value* addr = EmitBufferPointer(slice, target_shape); in EmitTargetAddressForOp()
2965 llvm::Value* op_target_address = EmitBufferPointer(slice, element_shape); in EmitTargetElementLoop()
3102 return EmitBufferPointer(root_buffer, root_inst->shape()); in GetBufferForGlobalCallReturnValue()
|
| D | ir_emitter.h | 270 llvm::Value* EmitBufferPointer(const BufferAllocation::Slice& slice,
|