Searched refs:top_diff_mat (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 564 ConstEigenMatrixMap top_diff_mat( in launch() local
576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
635 bottom_diff_ref = top_diff_mat.coeffRef(d, in_index); in launch()
|
| D | maxpooling_op.cc | 554 ConstEigenMatrixMap top_diff_mat( in SpatialMaxPoolGradGrad() local
581 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
626 bottom_diff_ref = top_diff_mat.coeffRef(d, in_index); in SpatialMaxPoolGradGrad()
|