Searched defs:PoolParameters (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dpooling_ops_common.cc31 PoolParameters::PoolParameters(OpKernelContext* context, function in class:tensorflow::PoolParameters
101 TensorShape PoolParameters::forward_output_shape() {
154 PoolParameters params{context, size, stride,
257 PoolParameters params{context, size, stride,
H A Dpooling_ops_common.h41 struct PoolParameters { struct in namespace:tensorflow
43 PoolParameters(OpKernelContext* context, const std::vector<int32>& ksize,
111 PoolParameters params{context, ksize_, stride_,
147 const Tensor& tensor_in, const PoolParameters& params) {
157 const Tensor& tensor_in, const PoolParameters& params,
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
[all...]

Completed in 112 milliseconds