| /external/tensorflow/tensorflow/core/kernels/ |
| pooling_ops_common.h | 211 const int32 output_image_size = out_height * out_width * params.depth;
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
213 1, (limit - start) * output_image_size);
458 const int32 output_image_size = out_height * out_width * params.depth;
459 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
460 1, (limit - start) * output_image_size);
527 const int64 output_image_size =
535 out_flat.data() + start * output_image_size, params.depth,
|
| conv_grad_input_ops.cc | 417 const int output_image_size = variable
428 const size_t size_A = output_image_size * dims.out_depth;
432 const size_t size_C = output_image_size * filter_total_size;
465 static_cast<int64>(output_image_size),
500 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
503 output_image_size, dims.out_depth);
532 &output_image_size, &filter_total_size,
542 MatrixMap C(im2col_buf, output_image_size, filter_total_size);
544 ConstMatrixMap A(out_data, output_image_size, dims.out_depth);
[all...] |
| depthwise_conv_grad_op.cc | 408 const int64 output_image_size =
437 out_backprop + b * output_image_size, out_bprop_buf);
659 const int64 output_image_size = local
[all...] |
| conv_grad_filter_ops.cc | 394 const int output_image_size = variable
408 const size_t size_A = output_image_size * filter_total_size;
410 const size_t size_B = output_image_size * dims.out_depth;
424 static_cast<int64>(output_image_size),
482 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit,
484 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit,
[all...] |
| depthwise_conv_op.cc | 191 const int64 output_image_size =
209 const int64 out_base = b * output_image_size;
|
| maxpooling_op.cc | 106 const int32 output_image_size = out_height * out_width * depth;
107 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1,
108 (limit - start) * output_image_size);
111 out_arg_max_mat.data() + start * output_image_size, 1,
112 (limit - start) * output_image_size);
583 const int32 output_image_size = out_height * out_width * params.depth;
585 bottom_diff_mat.data() + start * output_image_size, 1,
586 (limit - start) * output_image_size);
[all...] |
| pooling_ops_3d.cc | 597 const int32 output_image_size =
600 bottom_diff_mat.data() + start * output_image_size, 1,
601 (limit - start) * output_image_size);
[all...] |
| deep_conv2d.cc | [all...] |
