Searched refs:bottom_diff_mat (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 564 EigenMatrixMap bottom_diff_mat( in launch() local
572 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
596 bottom_diff_mat.data() + start * output_image_size, 1, in launch()
630 bottom_diff_mat.coeffRef(d, out_index); in launch()
|
| D | maxpooling_op.cc | 565 EigenMatrixMap bottom_diff_mat( in SpatialMaxPoolGradGrad() local
589 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
607 bottom_diff_mat.data() + start * output_image_size, 1, in SpatialMaxPoolGradGrad()
633 T& bottom_diff_ref = bottom_diff_mat.coeffRef(d, out_index); in SpatialMaxPoolGradGrad()
|