Searched refs:in_planes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 223 int64 in_planes = GetTensorDim(input, data_format, '0'); in launch() local
241 0, (out_planes - 1) * strides[0] + filter_planes - in_planes); in launch()
256 const uint64 m = in_batch * in_planes * in_rows * in_cols; in launch()
278 } else if (!is_grouped_convolution && filter_planes == in_planes && in launch()
284 const uint64 k = in_planes * in_rows * in_cols * in_depth; in launch()
319 const int64 new_in_planes = in_planes + planes_odd; in launch()
336 in_planes = new_in_planes; in launch()
358 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth); in launch()
402 .set_spatial_dim(DimIndex::Z, in_planes) in launch()
484 {{in_planes, in_rows, in_cols}}, in launch()
|
| D | pooling_ops_3d.cc | 575 const int32 in_planes = params.tensor_in_planes; in launch() local
608 const int p_end = std::min(p_start + window_planes, in_planes); in launch()
626 ((b * in_planes + p) * in_rows + h) * in_cols + w; in launch()
|
| D | fused_batch_norm_op.cc | 1272 int64 in_planes = GetTensorDim(x, tensor_format_, '0'); in ComputeWithReservedSpace() local
1277 {{in_planes, in_rows * in_cols}}, in_depth); in ComputeWithReservedSpace()
1433 int64 in_planes = GetTensorDim(x, tensor_format_, '0'); in ComputeWithReservedSpace() local
1438 {{in_planes, in_rows * in_cols}}, in_depth); in ComputeWithReservedSpace()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 79 const int in_planes = input.dim_size(1); in Compute() local
115 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 6 5-D Tensor with shape `[batch, in_planes, in_rows, in_cols, depth]`.
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2746 auto in_planes = c->Value(in_planes_dim); in __anon42d741193a02() local
2756 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anon42d741193a02()
|