Home
last modified time | relevance | path

Searched refs:out_mat (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_common.h217 EigenMatrixMap out_mat( in SpatialMaxPool()
235 auto shard = [&params, &in_mat, &out_mat](int64 start, int64 limit) { in SpatialMaxPool()
250 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()
278 out_mat.col(out_offset) = in SpatialMaxPool()
279 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset)); in SpatialMaxPool()
478 EigenMatrixMap out_mat(
496 auto shard = [&params, &in_mat, &out_mat](int64 start, int64 limit) {
511 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
539 out_mat.col(out_offset) =
540 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset));
[all …]
Dfractional_avg_pool_op.cc122 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()
Dgather_nd_op.h127 auto out_mat = out->shaped<T, 2>({N_result, slice_size}); in DoGatherNd() local
138 params_flat, indices_mat, out_mat); \ in DoGatherNd()
Dfractional_max_pool_op.cc128 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()
Dmaxpooling_op.cc88 EigenMatrixMap out_mat( in SpatialMaxPoolWithArgMaxHelper() local
108 auto shard = [&params, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
126 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
158 T& output_ref = out_mat.coeffRef(d, out_index); in SpatialMaxPoolWithArgMaxHelper()
559 ConstEigenMatrixMap out_mat( in SpatialMaxPoolGradGrad() local
589 auto shard = [&params, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
626 const T& output_ref = out_mat.coeffRef(d, out_index); in SpatialMaxPoolGradGrad()
Dpooling_ops_3d.cc557 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth, in launch() local
572 auto shard = [&params, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
620 const T& output_ref = out_mat.coeffRef(d, out_index); in launch()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1515 MatrixMap<float> out_mat(output_data, filter_rows, output_cols); in HybridConvPerChannel()
1526 out_mat.col(c) = in HybridConvPerChannel()
3190 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
3192 Eigen::VectorXf out_count(out_mat.cols()); in AveragePool()
3195 out_mat.setZero(); in AveragePool()
3215 out_mat.col(out_offset) += in AveragePool()
3225 out_mat.array().rowwise() /= out_count.transpose().array(); in AveragePool()
3387 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
3389 out_mat.setConstant(std::numeric_limits<float>::lowest()); in MaxPool()
3409 out_mat.col(out_offset) = in MaxPool()
[all …]