| /external/tensorflow/tensorflow/core/kernels/ |
| pooling_ops_common.h | 179 EigenMatrixMap out_mat(
197 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) {
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
240 out_mat.col(out_offset) =
241 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset));
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...] |
| fractional_avg_pool_op.cc | 122 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3],
125 Eigen::Matrix<T, Eigen::Dynamic, 1> out_count(out_mat.cols());
128 out_mat.setZero();
172 out_mat.col(out_offset) += in_mat.col(in_offset);
180 out_mat.array().rowwise() /= out_count.transpose().array();
|
| gather_nd_op.cc | 160 auto out_mat = out->shaped<T, 2>({N_result, slice_size}); local
171 params_flat, indices_mat, out_mat); \
|
| fractional_max_pool_op.cc | 128 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3],
175 out_mat.col(out_offset) =
176 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset));
|
| maxpooling_op.cc | 70 EigenMatrixMap out_mat(
90 auto shard = [¶ms, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop,
107 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1,
139 T& output_ref = out_mat.coeffRef(d, out_index);
537 ConstEigenMatrixMap out_mat(
567 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat](
604 const T& output_ref = out_mat.coeffRef(d, out_index);
[all...] |
| pooling_ops_3d.cc | 561 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth,
576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat](
624 const T& output_ref = out_mat.coeffRef(d, out_index);
[all...] |
| /external/libxcam/tests/ |
| test-video-stabilization.cpp | 260 cv::Mat out_mat; local
261 convert_to_mat (output_buf, out_mat);
262 writer.write (out_mat);
|
| test-image-stitching.cpp | 532 cv::Mat out_mat; local
533 convert_to_mat (output_buf, out_mat);
534 writer.write (out_mat);
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| optimized_ops.h | [all...] |
