Searched refs:tensor_in_cols (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 52 int tensor_in_cols; member
177 params.tensor_in_cols * params.tensor_in_rows * in SpatialMaxPool()
199 const int32 in_cols = params.tensor_in_cols; 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 *
446 const int32 in_cols = params.tensor_in_cols;
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 | 237 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in Init()
243 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Init()
259 tensor_in_cols = mklInputShape->GetSizes()[0]; in Init()
276 tensor_in_cols = mklInputShape->GetDimension('W'); in Init()
282 tensor_in_cols = mklInputShape->GetDimension3D('W'); in Init()
356 tensor_in_cols, window_cols, col_stride, in Init()
408 mkl_params->in_sizes[0] = params.tensor_in_cols; in ExtractMklOpParams()
|
| D | pooling_ops_common.cc | 63 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W'); in PoolParameters()
86 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in PoolParameters()
124 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth}); in forward_output_shape()
241 .set_width(params.tensor_in_cols) in Compute()
396 .set_width(params.tensor_in_cols) in Compute()
|
| D | maxpooling_op.cc | 79 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch); in SpatialMaxPoolWithArgMaxHelper()
105 const int32 in_cols = params.tensor_in_cols; 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()
585 const int32 in_cols = params.tensor_in_cols; in SpatialMaxPoolGradGrad()
732 params.depth, params.tensor_in_rows, params.tensor_in_cols, in Compute()
1045 params.tensor_in_cols, params.depth}); in Compute()
1305 params.tensor_in_cols, params.depth, params.out_height, in launch()
[all …]
|
| D | pooling_ops_3d.h | 53 int tensor_in_cols; member
|
| D | pooling_ops_3d.cc | 66 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2'); in Pool3dParameters()
91 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in Pool3dParameters()
558 params.tensor_in_planes * params.tensor_in_cols * in launch()
566 params.tensor_in_planes * params.tensor_in_cols * in launch()
581 const int32 in_cols = params.tensor_in_cols; in launch()
812 params.tensor_in_planes, params.tensor_in_rows, params.tensor_in_cols, in launch()
|
| D | mkl_pooling_ops_common.h | 368 int tensor_in_cols; member
399 tensor_in_cols(0), in MklPoolParameters()
|
| D | pooling_ops_3d_sycl.h | 72 in_cols_(params.tensor_in_cols), in SYCL3DPoolParams()
|