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
|