Home
last modified time | relevance | path

Searched defs:in_mat (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfractional_avg_pool_op.cc119 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
Dfractional_max_pool_op.cc125 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
Dmaxpooling_op.cc85 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper() local
556 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc553 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3189 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local
3386 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local
3550 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local