Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcopy_thunk.cc26 const BufferAllocation::Slice& destination_buffer, uint64 mem_size, in HostToDeviceCopyThunk() argument
30 destination_buffer_(destination_buffer), in HostToDeviceCopyThunk()
45 const BufferAllocation::Slice& destination_buffer, uint64 mem_size, in DeviceToDeviceCopyThunk() argument
49 destination_buffer_(destination_buffer), in DeviceToDeviceCopyThunk()
Dcopy_thunk.h37 const BufferAllocation::Slice& destination_buffer,
60 const BufferAllocation::Slice& destination_buffer,
Dnccl_all_reduce_thunk.h45 const BufferAllocation::Slice& destination_buffer,
Dnccl_all_reduce_thunk.cc347 const BufferAllocation::Slice& destination_buffer, in NcclAllReduceThunk() argument
353 destination_buffer_(destination_buffer) {} in NcclAllReduceThunk()
Dir_emitter_unnested.cc555 auto destination_buffer = GetAllocationSlice(*hlo); in HandleTriangularSolve() local
556 if (operand_buffer != destination_buffer) { in HandleTriangularSolve()
559 /*destination_buffer=*/destination_buffer, in HandleTriangularSolve()
1038 auto destination_buffer = GetAllocationSlice(*scatter); in HandleScatter() local
1039 if (operand_buffer != destination_buffer) { in HandleScatter()
1042 /*destination_buffer=*/destination_buffer, in HandleScatter()
1225 auto destination_buffer = GetAllocationSlice(*sort, shape_index); in HandleSort() local
1227 if (destination_buffer != source_address) { in HandleSort()
1232 /*destination_buffer=*/destination_buffer, in HandleSort()
/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.cc543 auto destination_buffer = GetAllocationSlice(*sort, shape_index); in HandleSort() local
545 EmitBufferPointer(destination_buffer, operand->shape()); in HandleSort()
547 if (destination_buffer != source_address) { in HandleSort()