Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.h294 const int filter_total_size, const int output_image_size,
297 MatrixMap C(im2col_buf, output_image_size, filter_total_size);
299 ConstMatrixMap A(out_data, output_image_size, dims_out_depth);
312 const int filter_total_size, const int output_image_size,
323 const int m = output_image_size;
339 im2col_buf, filter_total_size * output_image_size * sizeof(T));
614 const int output_image_size =
625 const size_t size_A = output_image_size * dims.out_depth;
629 const size_t size_C = output_image_size * filter_total_size;
662 static_cast<int64>(output_image_size),
[all …]
Dpooling_ops_common.h249 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPool() local
250 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()
251 1, (limit - start) * output_image_size); in SpatialMaxPool()
510 const int32 output_image_size = out_height * out_width * params.depth;
511 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
512 1, (limit - start) * output_image_size);
579 const int64 output_image_size =
587 out_flat.data() + start * output_image_size, params.depth,
Dconv_grad_ops_3d.cc391 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
402 const int64 size_A = output_image_size * dims.out_depth; in Compute()
406 const int64 size_C = output_image_size * filter_total_size; in Compute()
431 static_cast<int64>(output_image_size), in Compute()
494 TensorMap C(col_buffer_data, output_image_size, filter_total_size); in Compute()
497 output_image_size, dims.out_depth); in Compute()
538 &output_image_size, &filter_total_size, in Compute()
548 MatrixMap C(im2col_buf, output_image_size, filter_total_size); in Compute()
550 ConstMatrixMap A(out_data, output_image_size, dims.out_depth); in Compute()
858 const int64 output_image_size = dims.spatial_dims[0].output_size * in Compute() local
[all …]
Ddepthwise_conv_grad_op.cc423 const int64 output_image_size = in operator ()() local
452 out_backprop + b * output_image_size, out_bprop_buf); in operator ()()
801 const int64 output_image_size = in ComputeBackpropFilter() local
804 output_image_size - (filter_spatial_size * padded_out_depth_size); in ComputeBackpropFilter()
835 std::min(output_image_size, out_bprop_index + kPacketSize); in ComputeBackpropFilter()
912 const int64 output_image_size = in operator ()() local
930 out_backprop + b * output_image_size, in operator ()()
Dconv_grad_filter_ops.cc499 const int output_image_size = in Compute() local
513 const size_t size_A = output_image_size * filter_total_size; in Compute()
515 const size_t size_B = output_image_size * dims.out_depth; in Compute()
529 static_cast<int64>(output_image_size), in Compute()
587 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
589 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
Dmaxpooling_op.cc125 const int32 output_image_size = out_height * out_width * depth; in SpatialMaxPoolWithArgMaxHelper() local
126 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
127 (limit - start) * output_image_size); in SpatialMaxPoolWithArgMaxHelper()
130 out_arg_max_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
131 (limit - start) * output_image_size); in SpatialMaxPoolWithArgMaxHelper()
605 const int32 output_image_size = out_height * out_width * params.depth; in SpatialMaxPoolGradGrad() local
607 bottom_diff_mat.data() + start * output_image_size, 1, in SpatialMaxPoolGradGrad()
608 (limit - start) * output_image_size); in SpatialMaxPoolGradGrad()
Ddepthwise_conv_op.cc197 const int64 output_image_size = in operator ()() local
215 const int64 out_base = b * output_image_size; in operator ()()
Dpooling_ops_3d.cc593 const int32 output_image_size = in launch() local
596 bottom_diff_mat.data() + start * output_image_size, 1, in launch()
597 (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 ()()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dtranspose_conv.h42 const int output_image_size = output_height * output_width; in TransposeConvV2() local
48 const int output_offset = output_image_size * output_depth; in TransposeConvV2()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5729 const int output_image_size = output_height * output_width; in TransposeConvV2() local
5735 const int output_offset = output_image_size * output_depth; in TransposeConvV2()
5955 const int output_image_size = output_height * output_width; in TransposeConvV2() local
5961 const int output_offset = output_image_size * output_depth; in TransposeConvV2()