Searched refs:pooling_ratio_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fractional_avg_pool_op.cc | 40 OP_REQUIRES_OK(context, context->GetAttr("pooling_ratio", &pooling_ratio_)); in FractionalAvgPoolOp()
43 OP_REQUIRES(context, pooling_ratio_.size() == 4, in FractionalAvgPoolOp()
47 context, pooling_ratio_[0] == 1 || pooling_ratio_[3] == 1, in FractionalAvgPoolOp()
87 static_cast<int>(floor(input_size[i] / pooling_ratio_[i])); in Compute()
187 std::vector<float> pooling_ratio_; member in tensorflow::FractionalAvgPoolOp
|
| D | fractional_max_pool_op.cc | 40 OP_REQUIRES_OK(context, context->GetAttr("pooling_ratio", &pooling_ratio_)); in FractionalMaxPoolOp()
44 OP_REQUIRES(context, pooling_ratio_.size() == 4, in FractionalMaxPoolOp()
49 context, pooling_ratio_[0] == 1 || pooling_ratio_[3] == 1, in FractionalMaxPoolOp()
92 static_cast<int>(floor(input_size[i] / pooling_ratio_[i])); in Compute()
188 std::vector<float> pooling_ratio_; member in tensorflow::FractionalMaxPoolOp
|