Searched refs:tensor_in_rows (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 53 int tensor_in_rows; member
177 params.tensor_in_cols * params.tensor_in_rows * in SpatialMaxPool()
198 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPool()
252 params.tensor_in_rows * params.tensor_in_cols * params.depth; in SpatialMaxPool()
274 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
424 params.tensor_in_cols * params.tensor_in_rows *
445 const int32 in_rows = params.tensor_in_rows;
499 params.tensor_in_rows * params.tensor_in_cols * params.depth;
526 params.tensor_in_rows * params.tensor_in_cols * params.depth;
533 params.tensor_in_cols * params.tensor_in_rows * shard_batch_size);
[all …]
|
| D | mkl_pooling_ops_common.cc | 238 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in Init()
242 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Init()
260 tensor_in_rows = mklInputShape->GetSizes()[1]; in Init()
277 tensor_in_rows = mklInputShape->GetDimension('H'); in Init()
283 tensor_in_rows = mklInputShape->GetDimension3D('H'); in Init()
352 tensor_in_rows, window_rows, row_stride, in Init()
409 mkl_params->in_sizes[1] = params.tensor_in_rows; in ExtractMklOpParams()
|
| D | pooling_ops_common.cc | 64 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H'); in PoolParameters()
83 context, GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in PoolParameters()
124 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth}); in forward_output_shape()
240 .set_height(params.tensor_in_rows) in Compute()
395 .set_height(params.tensor_in_rows) in Compute()
|
| D | maxpooling_op.cc | 79 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
104 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolWithArgMaxHelper()
198 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols * in SpatialMaxPoolWithArgMaxHelper()
339 params.tensor_in_rows, params.tensor_in_cols, params.depth, in MaxPoolingBackwardCustomKernel()
550 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
556 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolGradGrad()
584 const int32 in_rows = params.tensor_in_rows; in SpatialMaxPoolGradGrad()
732 params.depth, params.tensor_in_rows, params.tensor_in_cols, in Compute()
1044 TensorShape out_shape({params.tensor_in_batch, params.tensor_in_rows, in Compute()
1304 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 | 65 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1'); in Pool3dParameters()
88 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride, in Pool3dParameters()
559 params.tensor_in_rows * in launch()
567 params.tensor_in_rows * params.tensor_in_batch); in launch()
580 const int32 in_rows = params.tensor_in_rows; in launch()
812 params.tensor_in_planes, params.tensor_in_rows, params.tensor_in_cols, in launch()
|
| D | mkl_pooling_ops_common.h | 369 int tensor_in_rows; member
400 tensor_in_rows(0), in MklPoolParameters()
|
| D | pooling_ops_3d_sycl.h | 71 in_rows_(params.tensor_in_rows), in SYCL3DPoolParams()
|