Searched refs:ConvolutionMetalSimd (Results 1 – 4 of 4) sorted by relevance
29 class ConvolutionMetalSimd : public GPUOperation {31 ConvolutionMetalSimd() = default;41 ConvolutionMetalSimd(ConvolutionMetalSimd&& kernel) = default;42 ConvolutionMetalSimd& operator=(ConvolutionMetalSimd&& kernel) = default;43 ConvolutionMetalSimd(const ConvolutionMetalSimd&) = delete;44 ConvolutionMetalSimd& operator=(const ConvolutionMetalSimd&) = delete;84 explicit ConvolutionMetalSimd(const OperationDef& definition) in ConvolutionMetalSimd() function86 friend ConvolutionMetalSimd CreateConvolutionMetalSimd(91 ConvolutionMetalSimd CreateConvolutionMetalSimd(
109 const ConvolutionMetalSimd::ConvParams& conv_params) { in GenerateConvolution()441 int3 ConvolutionMetalSimd::GetGridSize() const { in GetGridSize()454 ConvolutionMetalSimd CreateConvolutionMetalSimd( in CreateConvolutionMetalSimd()457 ConvolutionMetalSimd desc(definition); in CreateConvolutionMetalSimd()
67 ConvolutionMetalSimd conv = in SelectConvolutionApple()69 return std::make_unique<ConvolutionMetalSimd>(std::move(conv)); in SelectConvolutionApple()
220 ConvolutionMetalSimd operation_simd =223 src_tensor, std::make_unique<ConvolutionMetalSimd>(std::move(operation_simd)), dst_shape,281 ConvolutionMetalSimd operation_simd =284 src_tensor, std::make_unique<ConvolutionMetalSimd>(std::move(operation_simd)), dst_shape,318 ConvolutionMetalSimd operation_simd =320 auto op_ptr = std::make_unique<ConvolutionMetalSimd>(std::move(operation_simd));