Searched defs:out_mat (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op.h | 127 auto out_mat = out->shaped<T, 2>({N_result, slice_size}); in DoGatherNd() local
|
| D | fractional_avg_pool_op.cc | 127 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
|
| D | fractional_max_pool_op.cc | 137 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
|
| D | maxpooling_op.cc | 90 EigenMatrixMap out_mat( in SpatialMaxPoolWithArgMaxHelper() local
583 ConstEigenMatrixMap out_mat( in SpatialMaxPoolGradGrad() local
|
| D | pooling_ops_3d.cc | 580 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth, in launch() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 2870 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
3071 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
3235 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in L2Pool() local
|