HomeSort by relevance Sort by last modified time
    Searched defs:PoolParameters (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
pooling_ops_common.cc 31 PoolParameters::PoolParameters(OpKernelContext* context,
101 TensorShape PoolParameters::forward_output_shape() {
154 PoolParameters params{context, size, stride,
257 PoolParameters params{context, size, stride,
pooling_ops_common.h 41 struct PoolParameters {
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& params
    [all...]

Completed in 100 milliseconds