Searched refs:out_plane (Results 1 – 2 of 2) sorted by relevance
81 &out_plane, &pad_planes)); in Pool3dParameters()92 {{out_plane, out_height, out_width}}, depth); in forward_output_shape()558 params.out_plane * params.out_width * in launch()566 params.out_plane * params.out_width * params.out_height * in launch()587 const int32 out_plane = params.out_plane; in launch() local594 out_plane * out_height * out_width * params.depth; in launch()602 for (int pp = 0; pp < out_plane; ++pp) { in launch()617 ((b * out_plane + pp) * out_height + ph) * out_width + pw; in launch()644 params.out_plane * params.out_height * params.out_width * params.depth * in launch()806 tensor_out.flat<T>().data(), params.tensor_in_batch, params.out_plane, in launch()
67 int64 out_plane; member