HomeSort by relevance Sort by last modified time
    Searched refs:out_plane (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
pooling_ops_3d.h 67 int64 out_plane; member in struct:tensorflow::Pool3dParameters
pooling_ops_3d.cc 86 &out_plane, &pad_planes));
97 {{out_plane, out_height, out_width}}, depth);
562 params.out_plane * params.out_width *
570 params.out_plane * params.out_width * params.out_height *
591 const int32 out_plane = params.out_plane;
598 out_plane * out_height * out_width * params.depth;
606 for (int pp = 0; pp < out_plane; ++pp) {
621 ((b * out_plane + pp) * out_height + ph) * out_width + pw;
648 params.out_plane * params.out_height * params.out_width * params.depth
    [all...]
pooling_ops_3d_sycl.h 79 out_planes_(params.out_plane),
    [all...]

Completed in 261 milliseconds