Lines Matching defs:out_size
68 // from in_size to out_size.
77 gtl::ArraySlice<int64> in_size, gtl::ArraySlice<int64> out_size) {
78 CHECK_EQ(in_size.size(), out_size.size());
88 } else if (out_size[i] == 1) {
89 // If in_size[i] > 1 but out_size[i] == 1, then we slice out the first
94 static_cast<uint64>(out_size[i] - 1));
96 dims.kernel_size[i] = (out_size[i] - 1) / gcd;
145 std::vector<int64> out_size, const int64 channels) {
170 ComputeResizeConvolutionParameters(in_size, out_size);
184 if (in_size[i] == 1 && out_size[i] > 1) {
185 output = builder->Add(output, builder->ConstantR1<float>(out_size[i], 0),
275 std::vector<int64> out_size;
276 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(1, &out_size));
277 OP_REQUIRES(ctx, out_size.size() == 2,
279 out_size.size()));
280 OP_REQUIRES(ctx, out_size[0] > 0 && out_size[1] > 0,
282 out_size[0], ",", out_size[1], "]"));
288 // If in_size[i] > 1 and out_size[i] == 1, slice out the first input in
293 if (in_size[i] > 1 && out_size[i] == 1) {
294 // If in_size[i] > 1 but out_size[i] == 1, then we slice out the first
311 // while (out_size[0]-1) = c * 2^x * (in_size[0]-1) for x>1 c>1, resize the
322 while (in_size != out_size) {
325 (static_cast<float>(out_size[0]) - 1) / ((in_size[0] - 1) * 2),
326 (static_cast<float>(out_size[1]) - 1) / ((in_size[1] - 1) * 2)};
337 b, input, num_spatial_dims, in_size, out_size, channels);
338 in_size = out_size;
342 in_size, out_size, channels);
343 in_size = out_size;
