Home
last modified time | relevance | path

Searched refs:commandEncoder (Results 1 – 24 of 24) sorted by relevance

/external/pytorch/aten/src/ATen/native/mps/operations/
DBitwiseOps.mm190 id<MTLComputeCommandEncoder> commandEncoder = stream->commandEncoder();
192 …[commandEncoder pushDebugGroup:[NSString stringWithFormat:@"Dispatch %s kernel", kernel_name.c_str…
193 [commandEncoder setComputePipelineState:cplState];
194 mtl_setBytes(commandEncoder, length, 0);
195 mtl_setBuffer(commandEncoder, output, 1);
196 mtl_setBuffer(commandEncoder, self, 2);
197 mtl_setBuffer(commandEncoder, other, 3);
198 mtl_dispatch1DJob(commandEncoder, cplState, length);
220 id<MTLComputeCommandEncoder> commandEncoder = stream->commandEncoder();
222 …[commandEncoder pushDebugGroup:[NSString stringWithFormat:@"Dispatch %s kernel", kernel_name.c_str…
[all …]
DGamma.mm422 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
462 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
513 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DUnaryKernel.mm46 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DRenormKernel.mm65 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DCrossKernel.mm104 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DRepeat.mm127 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DMultiTensorApply.h245 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DQuantized.mm742 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
799 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
849 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DHistogramKernel.mm236 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DBucketization.mm219 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DBinaryKernel.mm271 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DView.mm805 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
861 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DLinearAlgebra.mm91 auto computeEncoder = stream->commandEncoder();
523 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
DIndexing.mm99 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
/external/mesa3d/src/vulkan/wsi/
Dwsi_common_metal_layer.m95 id<MTLBlitCommandEncoder> commandEncoder = [commandBuffer blitCommandEncoder];
103 [commandEncoder copyFromBuffer:image_buffer
112 [commandEncoder endEncoding];
/external/executorch/backends/apple/mps/runtime/
DMPSStream.h64 id<MTLComputeCommandEncoder> commandEncoder();
DMPSStream.mm50 id<MTLComputeCommandEncoder> MPSStream::commandEncoder() {
/external/pytorch/aten/src/ATen/mps/
DMPSStream.h64 MTLComputeCommandEncoder_t commandEncoder();
DMPSStream.mm54 id<MTLComputeCommandEncoder> MPSStream::commandEncoder() {
/external/pytorch/aten/src/ATen/native/mps/
DOperationUtils.mm762 id<MTLBuffer> generateKernelDataOffsets(id<MTLComputeCommandEncoder> commandEncoder,
789 [commandEncoder setComputePipelineState:kernelDataOffsetsPSO];
790 [commandEncoder setBytes:strides.data() length:sizeof(uint32_t) * nDim * nOffsets atIndex:0];
791 [commandEncoder setBuffer:kernelDataOffsets offset:0 atIndex:1];
792 …[commandEncoder setBytes:iterShapeData.data() length:sizeof(uint32_t) * iterShape.size() atIndex:2…
793 [commandEncoder setBytes:&nDim length:sizeof(uint32_t) atIndex:3];
795 mtl_dispatch1DJob(commandEncoder, kernelDataOffsetsPSO, numThreads);
DOperationUtils.h400 id<MTLBuffer> generateKernelDataOffsets(id<MTLComputeCommandEncoder> commandEncoder, const TensorIt…
/external/pytorch/test/cpp_extensions/
Dmps_extension.mm51 id<MTLComputeCommandEncoder> computeEncoder = mpsStream->commandEncoder();
/external/angle/src/libANGLE/renderer/wgpu/
DVertexArrayWgpu.cpp522 wgpu::CommandEncoder &commandEncoder = contextWgpu->getCurrentCommandEncoder(); in syncClientArrays() local
526 commandEncoder.CopyBufferToBuffer(copy.src->getBuffer(), copy.sourceOffset, in syncClientArrays()
Dwgpu_helpers.cpp614 wgpu::CommandEncoder &commandEncoder = context->getCurrentCommandEncoder(); in readDataImmediate() local
618 commandEncoder.CopyBufferToBuffer(getBuffer(), safeCopyOffset, result->buffer.getBuffer(), 0, in readDataImmediate()