pooling_ops_3d_sycl.h | 33 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, 41 in_planes_(in_planes), 57 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, 63 : SYCL3DPoolParams(depth, batch, in_planes, in_rows, in_cols, 124 MaxPool3DSYCL(const int depth, const int batch, const int in_planes, 132 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows, 190 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local 207 MaxPool3DSYCL<T> max_pool(depth, batch, in_planes, in_rows, in_cols, 235 MaxPool3DGradSYCL(const int depth, const int batch, const int in_planes, 245 : p_(depth, batch, in_planes, in_rows, in_cols, output_shape, window 356 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local 596 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local 728 const int in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); local [all...] |