Searched defs:PoolParameters (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | pooling_ops_common.cc | 31 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 D | pooling_ops_common.h | 41 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