Searched refs:ConstEigenMatrixMap (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fractional_max_pool_op.cc | 72 ConstEigenMatrixMap; in Compute() typedef
125 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute()
224 ConstEigenMatrixMap; in Compute() typedef
259 ConstEigenMatrixMap tensor_in_mat( in Compute()
326 ConstEigenMatrixMap tensor_out_mat( in Compute()
|
| D | fractional_avg_pool_op.cc | 69 ConstEigenMatrixMap; in Compute() typedef
119 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), input_size[3], in Compute()
229 ConstEigenMatrixMap; in Compute() typedef
275 ConstEigenMatrixMap out_backprop_mat(out_backprop.flat<T>().data(), in Compute()
|
| D | pooling_ops_common.h | 172 ConstEigenMatrixMap; in SpatialMaxPool() typedef
176 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in SpatialMaxPool()
419 ConstEigenMatrixMap;
423 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth,
516 ConstEigenMatrixMap;
531 ConstEigenMatrixMap in_mat(
|
| D | pooling_ops_3d.cc | 553 ConstEigenMatrixMap; in launch() typedef
557 ConstEigenMatrixMap in_mat(tensor_in.flat<T>().data(), params.depth, in launch()
561 ConstEigenMatrixMap out_mat(tensor_out.flat<T>().data(), params.depth, in launch()
564 ConstEigenMatrixMap top_diff_mat( in launch()
|
| D | maxpooling_op.cc | 71 ConstEigenMatrixMap; in SpatialMaxPoolWithArgMaxHelper() typedef
77 ConstEigenMatrixMap in_mat( in SpatialMaxPoolWithArgMaxHelper()
544 ConstEigenMatrixMap; in SpatialMaxPoolGradGrad() typedef
548 ConstEigenMatrixMap in_mat( in SpatialMaxPoolGradGrad()
551 ConstEigenMatrixMap out_mat( in SpatialMaxPoolGradGrad()
554 ConstEigenMatrixMap top_diff_mat( in SpatialMaxPoolGradGrad()
|