Searched refs:PoolParameters (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 41 struct PoolParameters { struct
43 PoolParameters(OpKernelContext* context, const std::vector<int32>& ksize,
111 PoolParameters params{context, ksize_, stride_, in Compute()
147 const Tensor& tensor_in, const PoolParameters& params) { in DepthwiseMaxPool()
157 const Tensor& tensor_in, const PoolParameters& params, in SpatialMaxPool()
270 static void launch(OpKernelContext* context, const PoolParameters& params,
350 PoolParameters params{context, ksize, stride,
386 const Tensor& tensor_in, const PoolParameters& params) {
396 const Tensor& tensor_in, const PoolParameters& params,
513 const Tensor& input, const PoolParameters& params,
|
| D | maxpooling_op.cc | 61 const PoolParameters& params, const bool include_batch_in_index) { in SpatialMaxPoolWithArgMaxHelper()
298 PoolParameters params{context, ksize, stride, in Compute()
331 PoolParameters params{context, size, stride, in MaxPoolingBackwardCustomKernel()
527 PoolParameters params{context, ksize, stride, in Compute()
541 const PoolParameters& params, in SpatialMaxPoolGradGrad()
726 PoolParameters params{context, ksize, stride, in Compute()
783 PoolParameters params{context, ksize_, stride_, in Compute()
861 PoolParameters params{context, ksize, stride, in Compute()
888 static void launch(OpKernelContext* context, const PoolParameters& params, in launch()
924 PoolParameters params{context, ksize_, stride_, in Compute()
[all …]
|
| D | pooling_ops_common.cc | 46 PoolParameters::PoolParameters(OpKernelContext* context, in PoolParameters() function in tensorflow::PoolParameters
116 TensorShape PoolParameters::forward_output_shape() { in forward_output_shape()
168 PoolParameters params{context, size, stride, in Compute()
303 PoolParameters params{context, size, stride, in Compute()
|
| D | quantized_pooling_ops.cc | 57 PoolParameters params{context, ksize_, stride_, in Compute()
|
| D | avgpooling_op.cc | 78 PoolParameters params{context, ksize_, stride_, in Compute()
143 PoolParameters params{context, ksize_, stride_, in Compute()
|