Searched defs:in_mat (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fractional_avg_pool_op.cc | 119 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
|
| D | fractional_max_pool_op.cc | 125 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
|
| D | maxpooling_op.cc | 77 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper() local
548 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad() local
|
| D | pooling_ops_3d.cc | 557 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3980 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/ |
| D | optimized_ops.h | 4127 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
|