HomeSort by relevance Sort by last modified time
    Searched defs:in_planes (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_ops_3d.cc 176 int64 in_planes = GetTensorDim(input, data_format, '0'); local
193 0, (out_planes - 1) * strides[0] + filter_planes - in_planes);
205 const uint64 m = in_batch * in_planes * in_rows * in_cols;
227 } else if (filter_planes == in_planes && filter_rows == in_rows &&
233 const uint64 k = in_planes * in_rows * in_cols * in_depth;
268 const int64 new_in_planes = in_planes + planes_odd;
285 in_planes = new_in_planes;
291 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth);
317 .set_spatial_dim(DimIndex::Z, in_planes)
377 {{in_planes, in_rows, in_cols}}
    [all...]
pooling_ops_3d_sycl.h 33 SYCL3DPoolParams(const int depth, const int batch, const int in_planes,
41 in_planes_(in_planes),
57 SYCL3DPoolParams(const int depth, const int batch, const int in_planes,
63 : SYCL3DPoolParams(depth, batch, in_planes, in_rows, in_cols,
124 MaxPool3DSYCL(const int depth, const int batch, const int in_planes,
132 : p_(depth, batch, in_planes, in_rows, in_cols, out_planes, out_rows,
190 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local
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
356 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local
596 const int in_planes = GetTensorDim(tensor_in, data_format, '0'); local
728 const int in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); local
    [all...]

Completed in 100 milliseconds