Searched refs:getPipelineStateForFunc (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | UnaryKernel.mm | 33 cplState = lib.getPipelineStateForFunc(name + "_complex_kernel", {scalarStr, scalarStr});
35 cplState = lib.getPipelineStateForFunc(name + "_kernel",
|
| D | FusedOptimizerOps.h | 441 return std::make_pair(lib.getPipelineStateForFunc(fname), lib.getMTLFunction(fname)); in getCPLState()
|
| D | RenormKernel.mm | 66 id<MTLComputePipelineState> renormPSO = lib.getPipelineStateForFunc(key);
|
| D | CrossKernel.mm | 107 auto crossPSO = lib.getPipelineStateForFunc("cross_" + scalarToMetalTypeString(out));
|
| D | HistogramKernel.mm | 247 …id<MTLComputePipelineState> stridedIndicesPSO = lib.getPipelineStateForFunc("kernel_index_offset");
258 id<MTLComputePipelineState> histogramPSO = lib.getPipelineStateForFunc(kernel);
|
| D | Repeat.mm | 128 …id<MTLComputePipelineState> pipelineState = lib.getPipelineStateForFunc("repeat_interleave", {scal…
|
| D | BitwiseOps.mm | 171 return lib.getPipelineStateForFunc(fname, {getMetalType(t1), getMetalType(t2), getMetalType(t3)});
|
| D | Quantized.mm | 744 id<MTLComputePipelineState> quantizedPSO = lib.getPipelineStateForFunc(kernel);
801 id<MTLComputePipelineState> quantizedPSO = lib.getPipelineStateForFunc(kernel);
857 id<MTLComputePipelineState> quantizedPSO = lib.getPipelineStateForFunc(kernel);
|
| D | Gamma.mm | 395 …return lib.getPipelineStateForFunc(fname, {scalarToMetalTypeString(t1), scalarToMetalTypeString(t2…
|
| D | Bucketization.mm | 223 id<MTLComputePipelineState> bucketizationPSO = lib.getPipelineStateForFunc(kernel);
|
| D | BinaryKernel.mm | 275 id<MTLComputePipelineState> binaryPSO = lib.getPipelineStateForFunc(kernel);
|
| D | View.mm | 782 …return (needsScatter ? scatterLib : gatherLib).getPipelineStateForFunc(kernel, {dtypeSrc, dtypeDst…
|
| D | LinearAlgebra.mm | 87 …auto matmulPSO = lib.getPipelineStateForFunc("naive_matmul_" + mps::scalarToMetalTypeString(output…
|
| /external/pytorch/aten/src/ATen/native/mps/ |
| D | OperationUtils.h | 345 inline id<MTLComputePipelineState> getPipelineStateForFunc(const std::string& fname) { in getPipelineStateForFunc() function
348 …id<MTLComputePipelineState> getPipelineStateForFunc(const std::string& fname, const std::initializ… in getPipelineStateForFunc() function
|