Home
last modified time | relevance | path

Searched defs:in_mat (Results 1 – 6 of 6) 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.cc77 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper() local
548 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc557 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3980 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local
4171 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local
4335 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local
4437 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in Softmax() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4127 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local
4302 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local
4450 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local
4552 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in Softmax() local