Searched refs:out_planes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | extract_volume_patches_op.cc | 109 int64 out_planes = 0, out_rows = 0, out_cols = 0; in Compute() local
113 padding_, &out_planes, &pad_planes)); in Compute()
122 batch, out_planes, out_rows, out_cols, in Compute()
|
| D | pooling_ops_3d_sycl.h | 34 const int in_rows, const int in_cols, const int out_planes, in SYCL3DPoolParams()
50 out_planes_(out_planes), in SYCL3DPoolParams()
125 const int in_rows, const int in_cols, const int out_planes, in MaxPool3DSYCL() argument
132 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows, in MaxPool3DSYCL()
186 const int out_planes = GetTensorDim(*output, data_format, '0');
208 out_planes, out_rows, out_cols, window, stride,
531 const int in_rows, const int in_cols, const int out_planes,
538 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows,
592 const int out_planes = GetTensorDim(*output, data_format, '0');
614 out_planes, out_rows, out_cols, window, stride,
|
| D | mkl_conv_ops.h | 390 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
416 padding_, &out_planes, &pad_D1, &pad_D2));
450 {{out_planes, out_rows, out_cols}}, out_depth);
465 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = static_cast<int>(out_planes);
|
| D | conv_ops_3d.cc | 225 int64 out_planes = GetTensorDim(*output, data_format, '0'); in launch() local
231 0, (out_planes - 1) * strides[0] + filter_planes - in_planes); in launch()
362 .set_spatial_dim(DimIndex::Z, out_planes) in launch()
400 {{out_planes, out_rows, out_cols}}, out_depth), in launch()
|
| D | eigen_cuboid_convolution.h | 1694 TensorIndex out_planes;
1699 out_planes = Eigen::divup(inputPlanes - kernelPlanes + 1,
1707 out_planes =
1714 out_planes = 0;
1737 pre_contract_dims[1] = out_planes * out_height * out_width;
1744 pre_contract_dims[0] = out_planes * out_height * out_width;
1763 post_contract_dims[1] = out_planes;
1771 post_contract_dims[NumDims - 2] = out_planes;
|
| D | mkl_pooling_ops_common.h | 382 int64 out_planes; // Pool3D member
410 out_planes(0), in MklPoolParameters()
502 static_cast<int>(mkl_pool_params.out_planes), in GetOutputDims()
|
| D | mkl_pooling_ops_common.cc | 348 &out_planes, &pad_P1, &pad_P2)); in Init()
365 FastBoundsCheck(out_planes, std::numeric_limits<int>::max()), in Init()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 12 5-D Tensor with shape `[batch, out_planes, out_rows, out_cols,
15 in the "depth" dimension. Note `out_planes`, `out_rows` and `out_cols`
|