/external/tensorflow/tensorflow/core/kernels/ |
D | conv_grad_input_ops.h | 294 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 …]
|
D | pooling_ops_common.h | 249 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,
|
D | conv_grad_ops_3d.cc | 391 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 …]
|
D | depthwise_conv_grad_op.cc | 423 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 ()()
|
D | conv_grad_filter_ops.cc | 499 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()
|
D | maxpooling_op.cc | 125 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()
|
D | depthwise_conv_op.cc | 197 const int64 output_image_size = in operator ()() local 215 const int64 out_base = b * output_image_size; in operator ()()
|
D | pooling_ops_3d.cc | 593 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()
|
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 ()()
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
D | transpose_conv.h | 42 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/ |
D | optimized_ops.h | 5729 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()
|