Searched refs:Pool3dParameters (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.h | 40 struct Pool3dParameters { struct
42 Pool3dParameters(OpKernelContext* context, const std::vector<int32>& ksize,
|
| D | pooling_ops_3d.cc | 53 Pool3dParameters::Pool3dParameters(OpKernelContext* context, in Pool3dParameters() function in tensorflow::Pool3dParameters
95 TensorShape Pool3dParameters::forward_output_shape() { in forward_output_shape()
543 static void launch(OpKernelContext* context, const Pool3dParameters& params, in launch()
698 Pool3dParameters params{context, ksize_, stride_, in Compute()
804 static void launch(OpKernelContext* context, const Pool3dParameters& params, in launch()
|
| D | pooling_ops_3d_sycl.h | 67 SYCL3DPoolParams(const Pool3dParameters& params) in SYCL3DPoolParams()
412 MaxPool3DGradGradSYCL(const Pool3dParameters& params,
476 static void launch(OpKernelContext* context, const Pool3dParameters& params,
|