| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 179 EigenMatrixMap out_mat( in SpatialMaxPool()
197 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) { in SpatialMaxPool()
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()
240 out_mat.col(out_offset) = in SpatialMaxPool()
241 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset)); in SpatialMaxPool()
426 EigenMatrixMap out_mat(
444 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) {
459 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
487 out_mat.col(out_offset) =
488 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset));
[all …]
|
| D | fractional_avg_pool_op.cc | 122 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
125 Eigen::Matrix<T, Eigen::Dynamic, 1> out_count(out_mat.cols()); in Compute()
128 out_mat.setZero(); in Compute()
172 out_mat.col(out_offset) += in_mat.col(in_offset); in Compute()
180 out_mat.array().rowwise() /= out_count.transpose().array(); in Compute()
|
| D | fractional_max_pool_op.cc | 128 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
175 out_mat.col(out_offset) = in Compute()
176 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset)); in Compute()
|
| D | gather_nd_op.cc | 160 auto out_mat = out->shaped<T, 2>({N_result, slice_size}); in DoGatherNd() local
171 params_flat, indices_mat, out_mat); \ in DoGatherNd()
|
| D | maxpooling_op.cc | 80 EigenMatrixMap out_mat( in SpatialMaxPoolWithArgMaxHelper() local
100 auto shard = [¶ms, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
118 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
150 T& output_ref = out_mat.coeffRef(d, out_index); in SpatialMaxPoolWithArgMaxHelper()
551 ConstEigenMatrixMap out_mat( in SpatialMaxPoolGradGrad() local
581 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
618 const T& output_ref = out_mat.coeffRef(d, out_index); in SpatialMaxPoolGradGrad()
|
| D | pooling_ops_3d.cc | 561 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth, in launch() local
576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
624 const T& output_ref = out_mat.coeffRef(d, out_index); in launch()
|
| /external/libxcam/tests/ |
| D | test-video-stabilization.cpp | 260 cv::Mat out_mat; in main() local
261 convert_to_mat (output_buf, out_mat); in main()
262 writer.write (out_mat); in main()
|
| D | test-image-stitching.cpp | 532 cv::Mat out_mat; in main() local
533 convert_to_mat (output_buf, out_mat); in main()
534 writer.write (out_mat); in main()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3981 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
3983 Eigen::VectorXf out_count(out_mat.cols()); in AveragePool()
3986 out_mat.setZero(); in AveragePool()
4006 out_mat.col(out_offset) += in AveragePool()
4016 out_mat.array().rowwise() /= out_count.transpose().array(); in AveragePool()
4172 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
4174 out_mat.setConstant(std::numeric_limits<float>::lowest()); in MaxPool()
4194 out_mat.col(out_offset) = in MaxPool()
4195 out_mat.col(out_offset) in MaxPool()
4336 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in L2Pool() local
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4128 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
4130 Eigen::VectorXf out_count(out_mat.cols()); in AveragePool()
4133 out_mat.setZero(); in AveragePool()
4153 out_mat.col(out_offset) += in AveragePool()
4163 out_mat.array().rowwise() /= out_count.transpose().array(); in AveragePool()
4303 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
4305 out_mat.setConstant(std::numeric_limits<float>::lowest()); in MaxPool()
4325 out_mat.col(out_offset) = in MaxPool()
4326 out_mat.col(out_offset) in MaxPool()
4451 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in L2Pool() local
[all …]
|