Home
last modified time | relevance | path

Searched refs:output_image_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc198 const int filter_total_size, const int output_image_size, in operator ()()
201 MatrixMap C(im2col_buf, output_image_size, filter_total_size); in operator ()()
203 ConstMatrixMap A(out_data, output_image_size, dims_out_depth); in operator ()()
216 const int filter_total_size, const int output_image_size, in operator ()()
236 const int n = output_image_size; in operator ()()
251 im2col_buf, filter_total_size * output_image_size * sizeof(T)); in operator ()()
396 const int output_image_size = in Compute() local
407 const size_t size_A = output_image_size * dims.out_depth; in Compute()
411 const size_t size_C = output_image_size * filter_total_size; in Compute()
444 static_cast<int64>(output_image_size), in Compute()
[all …]
Dpooling_ops_common.h211 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPool() local
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()
213 1, (limit - start) * output_image_size); in SpatialMaxPool()
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,
Dconv_grad_ops_3d.cc380 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
391 const int64 size_A = output_image_size * dims.out_depth; in Compute()
395 const int64 size_C = output_image_size * filter_total_size; in Compute()
420 static_cast<int64>(output_image_size), in Compute()
483 TensorMap C(col_buffer_data, output_image_size, filter_total_size); in Compute()
486 output_image_size, dims.out_depth); in Compute()
527 &output_image_size, &filter_total_size, in Compute()
537 MatrixMap C(im2col_buf, output_image_size, filter_total_size); in Compute()
539 ConstMatrixMap A(out_data, output_image_size, dims.out_depth); in Compute()
847 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
[all …]
Ddepthwise_conv_grad_op.cc411 const int64 output_image_size = in operator ()() local
440 out_backprop + b * output_image_size, out_bprop_buf); in operator ()()
760 const int64 output_image_size = in ComputeBackpropFilter() local
763 output_image_size - (filter_spatial_size * padded_out_depth_size); in ComputeBackpropFilter()
794 std::min(output_image_size, out_bprop_index + kPacketSize); in ComputeBackpropFilter()
871 const int64 output_image_size = in operator ()() local
889 out_backprop + b * output_image_size, in operator ()()
Dconv_grad_filter_ops.cc300 const int output_image_size = in Compute() local
314 const size_t size_A = output_image_size * filter_total_size; in Compute()
316 const size_t size_B = output_image_size * dims.out_depth; in Compute()
330 static_cast<int64>(output_image_size), in Compute()
388 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
390 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
Dmaxpooling_op.cc117 const int32 output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local
118 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
119 (limit - start) * output_image_size); in SpatialMaxPoolWithArgMaxHelper()
122 out_arg_max_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
123 (limit - start) * output_image_size); in SpatialMaxPoolWithArgMaxHelper()
597 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
599 bottom_diff_mat.data() + start * output_image_size, 1, in SpatialMaxPoolGradGrad()
600 (limit - start) * output_image_size); in SpatialMaxPoolGradGrad()
Ddepthwise_conv_op.cc192 const int64 output_image_size = in operator ()() local
210 const int64 out_base = b * output_image_size; in operator ()()
Dpooling_ops_3d.cc597 const int32 output_image_size = in launch() local
600 bottom_diff_mat.data() + start * output_image_size, 1, in launch()
601 (limit - start) * output_image_size); in launch()
Ddeep_conv2d.cc1115 const int64 output_image_size = args.out_rows * args.out_cols * out_depth; in operator ()() local
1122 const int64 out_base = b * output_image_size; in operator ()()