| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fractional_avg_pool_op_test.py | 114 def _ValidateFractionalAvgPoolResult(self, input_tensor, pooling_ratio,
123 pooling_ratio: A list or tuple of length 4, first and last element be 1.
133 pooling_ratio,
160 pooling_ratio = [1, math.sqrt(2), math.sqrt(2), 1]
164 pooling_ratio,
291 pooling_ratio = [1, 1.5, 1.5, 1]
296 pooling_ratio,
436 pooling_ratio = [1, math.sqrt(2), math.sqrt(3), 1]
444 pooling_ratio,
465 pooling_ratio = [1, math.sqrt(3), math.sqrt(2), 1
[all...] |
| fractional_max_pool_op_test.py | 114 def _ValidateFractionalMaxPoolResult(self, input_tensor, pooling_ratio,
123 pooling_ratio: A list or tuple of length 4, first and last element be 1.
133 pooling_ratio,
160 pooling_ratio = [1, math.sqrt(2), math.sqrt(2), 1]
164 pooling_ratio,
288 pooling_ratio = [1, 1.5, 1.5, 1]
293 pooling_ratio,
441 pooling_ratio = [1, math.sqrt(2), math.sqrt(3), 1]
449 pooling_ratio,
470 pooling_ratio = [1, math.sqrt(3), math.sqrt(2), 1
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| nn_ops_test.cc | 484 auto set_op = [&op, op_name](const std::vector<float>& pooling_ratio) {
487 .Attr("pooling_ratio", pooling_ratio)
506 // Wrong number of values for pooling_ratio.
508 INFER_ERROR("pooling_ratio field", op, "?");
510 INFER_ERROR("pooling_ratio field", op, "?");
|
| nn_ops.cc | 36 std::vector<float> pooling_ratio; local
37 TF_RETURN_IF_ERROR(c->GetAttr("pooling_ratio", &pooling_ratio));
38 if (pooling_ratio.size() != 4) {
40 "pooling_ratio field must specify 4 dimensions");
48 auto val = static_cast<int64>(floor(c->Value(d) / pooling_ratio[i]));
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_ops.py | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
