Searched refs:out_plane (Results 1 – 3 of 3) sorted by relevance
86 &out_plane, &pad_planes)); in Pool3dParameters()97 {{out_plane, out_height, out_width}}, depth); in forward_output_shape()562 params.out_plane * params.out_width * in launch()570 params.out_plane * params.out_width * params.out_height * in launch()591 const int32 out_plane = params.out_plane; in launch() local598 out_plane * out_height * out_width * params.depth; in launch()606 for (int pp = 0; pp < out_plane; ++pp) { in launch()621 ((b * out_plane + pp) * out_height + ph) * out_width + pw; in launch()648 params.out_plane * params.out_height * params.out_width * params.depth * in launch()810 tensor_out.flat<T>().data(), params.tensor_in_batch, params.out_plane, in launch()
67 int64 out_plane; member
79 out_planes_(params.out_plane), in SYCL3DPoolParams()