| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_pooling_test.cc | 422 const int in_c = c + k - dc; local
424 in_r < input_rows && in_c >= 0 && in_c < input_cols) {
425 expected_sum += input(d, in_p, in_r, in_c, b);
500 const int in_c = c + k - dc; local
502 in_r < input_rows && in_c >= 0 && in_c < input_cols) {
503 expected_sum += input(b, in_c, in_r, in_p, d);
|
| depthwise_conv_op.h | 228 const int64 in_c = in_c_start + f_c; local
230 if (in_r >= 0 && in_r < args.in_rows && in_c >= 0 &&
231 in_c < args.in_cols) {
232 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth;
|
| eigen_spatial_convolutions_test.cc | 1192 const int in_c = c - dc + k * stride; local
1263 const int in_c = c - dc + k * stride; local
[all...] |
| deep_conv2d.cc | 651 const int64 in_c = in_c_start + c; local
652 if (in_c < 0 || in_c >= args.in_cols) continue;
654 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth;
710 const int64 in_c = in_c_start + t * tile_stride_cols; local
711 CopyInputTile<T>()(args, transform, num_tiles, in_r, in_c, input,
747 const int64 num_tiles, const int64 in_r, const int64 in_c,
796 const int64 out_c_start = (in_c + t * tile_stride_cols) +
[all...] |
| depthwise_conv_grad_op.cc | 182 const int64 in_r, const int64 in_c,
195 // Calculate the output spatial region which used point (in_r, in_c) as input.
200 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride);
201 const int64 out_c_end = std::min(out_cols - 1, (in_c + pad_cols) / stride);
219 const int64 f_c = in_c + pad_cols - out_c * stride;
245 // result in 'output' at location computed from 'in_r' and 'in_c'.
278 const int64 in_r, const int64 in_c,
294 const int64 base_output_index = (in_r * args.in_cols + in_c) * in_depth;
436 for (int64 in_c = 0; in_c < args.in_cols; ++in_c)
962 const int in_c = in_c_start + f_c; local
[all...] |
