Searched refs:bottom_diff_mat (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 568 EigenMatrixMap bottom_diff_mat( in launch() local
576 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in launch()
600 bottom_diff_mat.data() + start * output_image_size, 1, in launch()
634 bottom_diff_mat.coeffRef(d, out_index); in launch()
|
| D | maxpooling_op.cc | 557 EigenMatrixMap bottom_diff_mat( in SpatialMaxPoolGradGrad() local
581 auto shard = [¶ms, &in_mat, &out_mat, &top_diff_mat, &bottom_diff_mat]( in SpatialMaxPoolGradGrad()
599 bottom_diff_mat.data() + start * output_image_size, 1, in SpatialMaxPoolGradGrad()
625 T& bottom_diff_ref = bottom_diff_mat.coeffRef(d, out_index); in SpatialMaxPoolGradGrad()
|