Home
last modified time | relevance | path

Searched defs:out_mat (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op.cc160 auto out_mat = out->shaped<T, 2>({N_result, slice_size}); in DoGatherNd() local
Dfractional_avg_pool_op.cc122 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
Dfractional_max_pool_op.cc128 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute() local
Dmaxpooling_op.cc80 EigenMatrixMap out_mat( in SpatialMaxPoolWithArgMaxHelper() local
551 ConstEigenMatrixMap out_mat( in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc561 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth, in launch() local
/external/libxcam/tests/
Dtest-video-stabilization.cpp260 cv::Mat out_mat; in main() local
Dtest-image-stitching.cpp532 cv::Mat out_mat; in main() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3981 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
4172 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
4336 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in L2Pool() local
4438 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in Softmax() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4128 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in AveragePool() local
4303 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in MaxPool() local
4451 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in L2Pool() local
4553 auto out_mat = MapAsMatrixWithLastDimAsRows(output_data, output_shape); in Softmax() local