Searched refs:in_mat (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 176 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in SpatialMaxPool()
197 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) { in SpatialMaxPool()
241 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset)); in SpatialMaxPool()
423 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth,
444 auto shard = [¶ms, &in_mat, &out_mat](int64 start, int64 limit) {
488 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset));
531 ConstEigenMatrixMap in_mat(
580 out_mat.col(out_offset) += in_mat.col(in_offset);
|
| D | fractional_avg_pool_op.cc | 119 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
172 out_mat.col(out_offset) += in_mat.col(in_offset); in Compute()
|
| D | fractional_max_pool_op.cc | 125 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
176 out_mat.col(out_offset).cwiseMax(in_mat.col(in_offset)); in Compute()
|
| D | maxpooling_op.cc | 77 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper() local
100 auto shard = [¶ms, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
149 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolWithArgMaxHelper()
548 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad() local
581 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
623 const T& input_ref = in_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad()
|
| D | pooling_ops_3d.cc | 557 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch() local
576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
631 const T& input_ref = in_mat.coeffRef(d, in_index); in launch()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3980 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local
4007 in_mat.col(NodeOffset(b, h, w, input_height, input_width)); in AveragePool()
4171 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local
4196 .cwiseMax(in_mat.col( in MaxPool()
4335 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local
4337 Eigen::VectorXf in_square(in_mat.rows()); in L2Pool()
4362 in_mat.col(in_offset).array() * in_mat.col(in_offset).array(); in L2Pool()
4437 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in Softmax() local
4442 (in_mat.rowwise() - in_mat.colwise().maxCoeff()).array() * params.beta; in Softmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4127 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local
4154 in_mat.col(NodeOffset(b, h, w, input_height, input_width)); in AveragePool()
4302 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local
4327 .cwiseMax(in_mat.col( in MaxPool()
4450 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local
4452 Eigen::VectorXf in_square(in_mat.rows()); in L2Pool()
4477 in_mat.col(in_offset).array() * in_mat.col(in_offset).array(); in L2Pool()
4552 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in Softmax() local
4557 (in_mat.rowwise() - in_mat.colwise().maxCoeff()).array() * params.beta; in Softmax()
|