| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_pooling_ops_common.cc | 39 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H');
55 tensor_in_rows = mklInputShape->GetSizes()[1];
70 tensor_in_rows = mklInputShape->GetDimension('H');
104 tensor_in_rows, window_rows, row_stride,
155 mkl_params->in_sizes[1] = params.tensor_in_rows;
|
| pooling_ops_3d.h | 54 int tensor_in_rows; member in struct:tensorflow::Pool3dParameters
|
| pooling_ops_common.h | 53 int tensor_in_rows; 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)
198 const int32 in_rows = params.tensor_in_rows;
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)
445 const int32 in_rows = params.tensor_in_rows;
499 params.tensor_in_rows * params.tensor_in_cols * params.depth
[all...] |
| pooling_ops_common.cc | 49 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, 'H');
68 context, GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride,
109 {tensor_in_batch, tensor_in_rows, tensor_in_cols, out_depth});
199 .set_height(params.tensor_in_rows)
349 .set_height(params.tensor_in_rows)
|
| 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)
93 const int32 in_rows = params.tensor_in_rows;
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)
570 const int32 in_rows = params.tensor_in_rows;
[all...] |
| pooling_ops_3d.cc | 65 tensor_in_rows = GetTensorDim(tensor_in_shape, data_format, '1');
88 GetWindowedOutputSize(tensor_in_rows, window_rows, row_stride,
559 params.tensor_in_rows *
567 params.tensor_in_rows * params.tensor_in_batch);
580 const int32 in_rows = params.tensor_in_rows;
[all...] |
| mkl_pooling_ops_common.h | 41 int tensor_in_rows; member in struct:tensorflow::MklPoolParameters
66 tensor_in_rows(0),
|
| pooling_ops_3d_sycl.h | 71 in_rows_(params.tensor_in_rows),
[all...] |
