Searched refs:EigenMatrixMap (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 174 EigenMatrixMap; in SpatialMaxPool() typedef
179 EigenMatrixMap out_mat( in SpatialMaxPool()
212 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, in SpatialMaxPool()
421 EigenMatrixMap;
426 EigenMatrixMap out_mat(
459 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size,
518 EigenMatrixMap;
534 EigenMatrixMap out_mat(
|
| D | fractional_max_pool_op.cc | 74 EigenMatrixMap; in Compute() typedef
128 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute()
226 EigenMatrixMap; in Compute() typedef
262 EigenMatrixMap tensor_out_dup_mat( in Compute()
|
| D | maxpooling_op.cc | 73 EigenMatrixMap; in SpatialMaxPoolWithArgMaxHelper() typedef
80 EigenMatrixMap out_mat( in SpatialMaxPoolWithArgMaxHelper()
118 EigenMatrixMap out_shard(out_mat.data() + start * output_image_size, 1, in SpatialMaxPoolWithArgMaxHelper()
177 EigenMatrixMap in_shard(input_backprop_flat.data() + in_start, 1, in SpatialMaxPoolWithArgMaxHelper()
546 EigenMatrixMap; in SpatialMaxPoolGradGrad() typedef
557 EigenMatrixMap bottom_diff_mat( in SpatialMaxPoolGradGrad()
598 EigenMatrixMap bottom_diff_shard( in SpatialMaxPoolGradGrad()
956 EigenMatrixMap; typedef
978 EigenMatrixMap inputShard(grad_out_flat.data() + output_start, 1, in launch()
|
| D | fractional_avg_pool_op.cc | 71 EigenMatrixMap; in Compute() typedef
122 EigenMatrixMap out_mat(output_tensor->flat<T>().data(), output_size[3], in Compute()
|
| D | pooling_ops_3d.cc | 555 EigenMatrixMap; in launch() typedef
568 EigenMatrixMap bottom_diff_mat( in launch()
599 EigenMatrixMap bottom_diff_shard( in launch()
|