| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.cc | 38 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W');
54 tensor_in_cols = mklInputShape->GetSizes()[0];
69 tensor_in_cols = mklInputShape->GetDimension('W');
108 tensor_in_cols, window_cols, col_stride,
154 mkl_params->in_sizes[0] = params.tensor_in_cols;
|
| pooling_ops_3d.h | 53 int tensor_in_cols; member in struct:tensorflow::Pool3dParameters
|
| pooling_ops_common.h | 52 int tensor_in_cols; member in struct:tensorflow::PoolParameters
177 params.tensor_in_cols * params.tensor_in_rows *
189 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
199 const int32 in_cols = params.tensor_in_cols;
252 params.tensor_in_rows * params.tensor_in_cols * params.depth;
274 params.tensor_in_batch, params.tensor_in_rows, params.tensor_in_cols,
424 params.tensor_in_cols * params.tensor_in_rows *
436 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
446 const int32 in_cols = params.tensor_in_cols;
499 params.tensor_in_rows * params.tensor_in_cols * params.depth
[all...] |
| pooling_ops_common.cc | 48 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, 'W');
71 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride,
109 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth});
200 .set_width(params.tensor_in_cols)
350 .set_width(params.tensor_in_cols)
|
| maxpooling_op.cc | 69 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch);
83 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
94 const int32 in_cols = params.tensor_in_cols;
184 const int64 shard_cost = params.tensor_in_rows * params.tensor_in_cols *
325 params.tensor_in_rows, params.tensor_in_cols, params.depth,
536 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch);
542 params.tensor_in_cols * params.tensor_in_rows * params.tensor_in_batch);
554 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
558 // depth by (tensor_in_cols * tensor_in_rows * tensor_in_batch)
571 const int32 in_cols = params.tensor_in_cols;
[all...] |
| pooling_ops_3d.cc | 66 tensor_in_cols = GetTensorDim(tensor_in_shape, data_format, '2');
91 GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride,
558 params.tensor_in_planes * params.tensor_in_cols *
566 params.tensor_in_planes * params.tensor_in_cols *
581 const int32 in_cols = params.tensor_in_cols;
[all...] |
| mkl_pooling_ops_common.h | 40 int tensor_in_cols; member in struct:tensorflow::MklPoolParameters
65 tensor_in_cols(0),
|
| pooling_ops_3d_sycl.h | 72 in_cols_(params.tensor_in_cols),
[all...] |
