Home
last modified time | relevance | path

Searched refs:out_planes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/image/
Dextract_volume_patches_op.cc112 int64 out_planes = 0, out_rows = 0, out_cols = 0; in Compute() local
116 padding_, &out_planes, &pad_planes)); in Compute()
125 batch, out_planes, out_rows, out_cols, in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExtractVolumePatches.pbtxt12 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`
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h396 int64 out_rows = 0, out_cols = 0, out_planes = 0; variable
422 stride_planes, padding_, &out_planes,
459 {{out_planes, out_rows, out_cols}}, out_depth);
474 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = static_cast<int>(out_planes);
Dmkl_pooling_ops_common.h375 int64 out_planes; // Pool3D member
403 out_planes(0), in MklPoolParameters()
492 static_cast<int>(mkl_pool_params.out_planes), in GetOutputDims()
563 static_cast<int>(pool_params->out_planes), in AllocateEmptyOutputTensor()
Dmkl_pooling_ops_common.cc333 &out_planes, &pad_P1, &pad_P2)); in Init()
349 FastBoundsCheck(out_planes, std::numeric_limits<int>::max()), in Init()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc235 int64 out_planes = GetTensorDim(*output, data_format, '0'); in launch() local
241 0, (out_planes - 1) * strides[0] + filter_planes - in_planes); in launch()
408 .set_spatial_dim(DimIndex::Z, out_planes) in launch()
461 {{out_planes, out_rows, out_cols}}, out_depth), in launch()
Deigen_cuboid_convolution.h1888 TensorIndex out_planes;
1893 out_planes = Eigen::divup(inputPlanes - kernelPlanes + 1,
1901 out_planes =
1908 out_planes = 0;
1931 pre_contract_dims[1] = out_planes * out_height * out_width;
1938 pre_contract_dims[0] = out_planes * out_height * out_width;
1957 post_contract_dims[1] = out_planes;
1965 post_contract_dims[NumDims - 2] = out_planes;