Searched defs:in_mat (Results 1 – 7 of 7) sorted by relevance
211 let in_mat = localVariable226 let in_mat = localVariable241 let in_mat = localVariable256 let in_mat = localVariable
368 let in_mat = localVariable383 let in_mat = localVariable398 let in_mat = localVariable413 let in_mat = localVariable
134 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
124 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute() local
87 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper() local580 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad() local
576 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch() local
2869 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in AveragePool() local3070 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in MaxPool() local3234 const auto in_mat = MapAsMatrixWithLastDimAsRows(input_data, input_shape); in L2Pool() local