Home
last modified time | relevance | path

Searched refs:in_planes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d_sycl.h33 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, in SYCL3DPoolParams()
41 in_planes_(in_planes), in SYCL3DPoolParams()
57 SYCL3DPoolParams(const int depth, const int batch, const int in_planes, in SYCL3DPoolParams()
63 : SYCL3DPoolParams(depth, batch, in_planes, in_rows, in_cols, in SYCL3DPoolParams()
124 MaxPool3DSYCL(const int depth, const int batch, const int in_planes, in MaxPool3DSYCL() argument
132 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows, in MaxPool3DSYCL()
190 const int in_planes = GetTensorDim(tensor_in, data_format, '0');
207 MaxPool3DSYCL<T> max_pool(depth, batch, in_planes, in_rows, in_cols,
235 MaxPool3DGradSYCL(const int depth, const int batch, const int in_planes,
245 : p_(depth, batch, in_planes, in_rows, in_cols, output_shape, window,
[all …]
Dconv_ops_3d.cc214 int64 in_planes = GetTensorDim(input, data_format, '0'); in launch() local
231 0, (out_planes - 1) * strides[0] + filter_planes - in_planes); in launch()
244 const uint64 m = in_batch * in_planes * in_rows * in_cols; in launch()
266 } else if (filter_planes == in_planes && filter_rows == in_rows && in launch()
272 const uint64 k = in_planes * in_rows * in_cols * in_depth; in launch()
307 const int64 new_in_planes = in_planes + planes_odd; in launch()
324 in_planes = new_in_planes; in launch()
330 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
356 .set_spatial_dim(DimIndex::Z, in_planes) in launch()
420 {{in_planes, in_rows, in_cols}}, in launch()
Dextract_volume_patches_op.cc76 const int in_planes = input.dim_size(1); in Compute() local
112 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
Dpooling_ops_3d.cc579 const int32 in_planes = params.tensor_in_planes; in launch() local
612 const int p_end = std::min(p_start + window_planes, in_planes); in launch()
630 ((b * in_planes + p) * in_rows + h) * in_cols + w; in launch()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExtractVolumePatches.pbtxt6 5-D Tensor with shape `[batch, in_planes, in_rows, in_cols, depth]`.
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc2587 auto in_planes = c->Value(in_planes_dim); in __anondb9326b23a02() local
2597 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anondb9326b23a02()