Searched refs:complex_kernel (Results 1 – 2 of 2) sorted by relevance
10 void complex_kernel(TensorIterator& iter) { in complex_kernel() function28 REGISTER_DISPATCH(complex_stub, &complex_kernel);
230 kernel void complex_kernel(constant void * real_ [[buffer(0)]],245 kernel void complex_kernel<DTYPE>( \371 mps::binary_mps_impl(iter, "complex_kernel");