Searched refs:output_image_size (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_grad_input_ops.cc | 198 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 …]
|
D | pooling_ops_common.h | 211 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,
|
D | conv_grad_ops_3d.cc | 380 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 …]
|
D | depthwise_conv_grad_op.cc | 411 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 ()()
|
D | conv_grad_filter_ops.cc | 300 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()
|
D | maxpooling_op.cc | 117 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()
|
D | depthwise_conv_op.cc | 192 const int64 output_image_size = in operator ()() local 210 const int64 out_base = b * output_image_size; in operator ()()
|
D | pooling_ops_3d.cc | 597 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()
|
D | deep_conv2d.cc | 1115 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 ()()
|