Searched refs:tensor_in_rows (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.cc | 137 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
166 tensor_in_rows, window_rows, row_stride, in PoolParameters()
209 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth}); in forward_output_shape()
241 int tensor_in_rows = params.tensor_in_rows; in Compute() local
314 const int64_t new_in_rows = tensor_in_rows + padding_rows_diff; in Compute()
351 tensor_in_rows = new_in_rows; in Compute()
367 .set_height(tensor_in_rows) in Compute()
579 int tensor_in_rows = params.tensor_in_rows; in Compute() local
604 const int64_t new_in_rows = tensor_in_rows + padding_rows_diff; in Compute()
608 << " original rows=" << tensor_in_rows in Compute()
[all …]
|
| D | pooling_ops_common.h | 61 int tensor_in_rows; member
215 params.tensor_in_cols * params.tensor_in_rows * in SpatialMaxPool()
236 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPool()
291 params.tensor_in_rows * params.tensor_in_cols * params.depth; in SpatialMaxPool()
315 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
477 params.tensor_in_cols * params.tensor_in_rows *
498 const int32_t in_rows = params.tensor_in_rows;
553 params.tensor_in_rows * params.tensor_in_cols * params.depth;
580 params.tensor_in_rows * params.tensor_in_cols * params.depth;
587 params.tensor_in_cols * params.tensor_in_rows * shard_batch_size);
[all …]
|
| D | maxpooling_op.cc | 89 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
114 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper()
211 const int64_t shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper()
562 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
568 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
596 const int32_t in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad()
749 params.depth, params.tensor_in_rows, params.tensor_in_cols, in Compute()
1100 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute()
1380 input.flat<T>().data(), params.tensor_in_batch, params.tensor_in_rows, in launch()
1399 input.flat<T>().data(), params.tensor_in_batch, params.tensor_in_rows, in launch()
[all …]
|
| D | pooling_ops_3d.h | 54 int tensor_in_rows; member
|
| D | pooling_ops_3d.cc | 60 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters()
83 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in Pool3dParameters()
568 params.tensor_in_rows * in launch()
576 params.tensor_in_rows * params.tensor_in_batch); in launch()
589 const int32_t in_rows = params.tensor_in_rows; in launch()
846 params.tensor_in_planes, params.tensor_in_rows, params.tensor_in_cols, in launch()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 239 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init()
243 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init()
262 tensor_in_rows = mklInputShape->GetDimension('H'); in Init()
268 tensor_in_rows = mklInputShape->GetDimension3D('H'); in Init()
337 tensor_in_rows, window_rows, row_stride, in Init()
|
| D | mkl_pooling_ops_common.h | 362 int tensor_in_rows; member
393 tensor_in_rows(0), in MklPoolParameters()
|