Home
last modified time | relevance | path

Searched refs:destination_buffer (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcopy_thunk.cc26 const BufferAllocation::Slice& destination_buffer, uint64 mem_size) in HostToDeviceCopyThunk() argument
29 destination_buffer_(destination_buffer), in HostToDeviceCopyThunk()
43 const BufferAllocation::Slice& destination_buffer, uint64 mem_size) in DeviceToDeviceCopyThunk() argument
46 destination_buffer_(destination_buffer), in DeviceToDeviceCopyThunk()
Dcopy_thunk.h37 const BufferAllocation::Slice& destination_buffer,
59 const BufferAllocation::Slice& destination_buffer,
Dnccl_all_to_all_thunk.cc86 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
118 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
Dnccl_collective_thunk.h112 BufferAllocation::Slice destination_buffer; member
Dnccl_all_gather_thunk.cc86 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
Dnccl_all_reduce_thunk.cc137 params.buffer_allocations->GetDeviceAddress(buffer.destination_buffer) in RunNcclCollective()
Dir_emitter_unnested.cc2098 auto destination_buffer = *GetAllocationSliceForMlir(copy.output()); in EmitCopyForMlir() local
2099 if (operand_buffer != destination_buffer) { in EmitCopyForMlir()
2103 /*destination_buffer=*/destination_buffer, in EmitCopyForMlir()
2874 TF_ASSIGN_OR_RETURN(auto destination_buffer, in EmitSortFromMlir()
2878 if (destination_buffer != source_address) { in EmitSortFromMlir()
2885 /*destination_buffer=*/destination_buffer, in EmitSortFromMlir()
3160 /*destination_buffer=*/buffers[i].destination_buffer, in EmitNcclThunkFromMlir()
/external/deqp/external/openglcts/modules/common/
DglcRobustBufferAccessBehaviorTests.cpp3234 Buffer destination_buffer(gl); in iterate() local
3239 destination_buffer.InitData(GL_SHADER_STORAGE_BUFFER, GL_DYNAMIC_COPY, sizeof(m_destination_data), in iterate()
3243 destination_buffer.BindBase(0); in iterate()
3262 destination_buffer.Bind(); in iterate()
3505 Buffer destination_buffer(gl); in iterate() local
3510 destination_buffer.InitData(GL_SHADER_STORAGE_BUFFER, GL_DYNAMIC_COPY, sizeof(destination_data), in iterate()
3514 destination_buffer.BindBase(0); in iterate()
3533 destination_buffer.Bind(); in iterate()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc568 auto destination_buffer = GetAllocationSlice(*sort, shape_index); in HandleSort() local
570 EmitBufferPointer(destination_buffer, operand->shape()); in HandleSort()
572 if (destination_buffer != source_address) { in HandleSort()