| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_grad_op.cc | 182 const int64 in_r, const int64 in_c, in CopyOutputBackpropRegion() argument
200 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride); in CopyOutputBackpropRegion()
201 const int64 out_c_end = std::min(out_cols - 1, (in_c + pad_cols) / stride); in CopyOutputBackpropRegion()
219 const int64 f_c = in_c + pad_cols - out_c * stride; in CopyOutputBackpropRegion()
278 const int64 in_r, const int64 in_c, in ComputeBackpropInput() argument
294 const int64 base_output_index = (in_r * args.in_cols + in_c) * in_depth; in ComputeBackpropInput()
436 for (int64 in_c = 0; in_c < args.in_cols; ++in_c) { in operator ()() local
439 args, padded_filter_inner_dim_size, in_r, in_c, in operator ()()
444 in_c, filter_data, out_bprop_buf, in operator ()()
467 for (int in_c = 0; in_c < args.in_cols; ++in_c) { in DepthwiseConvBackpropInputReference() local
[all …]
|
| D | deep_conv2d.cc | 651 const int64 in_c = in_c_start + c; in operator ()() local
652 if (in_c < 0 || in_c >= args.in_cols) continue; in operator ()()
654 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth; in operator ()()
710 const int64 in_c = in_c_start + t * tile_stride_cols; in operator ()() local
711 CopyInputTile<T>()(args, transform, num_tiles, in_r, in_c, input, in operator ()()
747 const int64 num_tiles, const int64 in_r, const int64 in_c, in operator ()()
796 const int64 out_c_start = (in_c + t * tile_stride_cols) + in operator ()()
881 const Conv2DState<T>& cs, const int64 in_r, const int64 in_c, in operator ()()
886 TransformInputTiles<T>()(args, transform, num_tiles, in_r, in_c, input, in operator ()()
913 TransformOutputTile<T>()(args, transform, num_tiles, in_r, in_c, in operator ()()
[all …]
|
| D | depthwise_conv_op.h | 228 const int64 in_c = in_c_start + f_c;
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;
|
| D | fractional_avg_pool_op.cc | 300 for (int64 in_c = in_col_start; in_c <= in_col_end; ++in_c) { in Compute() local
301 const int64 in_index = (b * in_rows + in_r) * in_cols + in_c; in Compute()
|
| D | eigen_pooling_test.cc | 422 const int in_c = c + k - dc; in TEST() local
424 in_r < input_rows && in_c >= 0 && in_c < input_cols) { in TEST()
425 expected_sum += input(d, in_p, in_r, in_c, b); in TEST()
500 const int in_c = c + k - dc; in TEST() local
502 in_r < input_rows && in_c >= 0 && in_c < input_cols) { in TEST()
503 expected_sum += input(b, in_c, in_r, in_p, d); in TEST()
|
| D | eigen_spatial_convolutions_test.cc | 1192 const int in_c = c - dc + k * stride; in TEST() local
1193 if (in_p >= 0 && in_r >= 0 && in_c >= 0 && in_p < in_depth && in TEST()
1194 in_r < in_rows && in_c < in_cols) { in TEST()
1196 input(id, in_p, in_r, in_c) * kernel(od, id, p, r, c); in TEST()
1263 const int in_c = c - dc + k * stride; in TEST() local
1264 if (in_p >= 0 && in_r >= 0 && in_c >= 0 && in_p < in_depth && in TEST()
1265 in_r < in_rows && in_c < in_cols) { in TEST()
1267 input(in_c, in_r, in_p, id) * kernel(c, r, p, id, od); in TEST()
|