Searched refs:spatial_dim_count (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | data_format_vec_permute.cc | 84 const int spatial_dim_count = get_spatial_dim_count(*src_format); in Validate() local
86 if (x_dims.d[0] != spatial_dim_count && x_dims.d[0] != full_dim_count) { in Validate()
88 spatial_dim_count, " or ", in Validate()
93 if (x_dims.d[0] != spatial_dim_count && x_dims.d[0] != full_dim_count) { in Validate()
95 "First dimension of 2D input must be of size ", spatial_dim_count, in Validate()
120 const int& spatial_dim_count = get_spatial_dim_count(src_format); in Convert() local
124 if (attrs_.x_dim_count == spatial_dim_count) { in Convert()
126 [spatial_dim_count](string* format_str) -> void { in Convert()
128 [spatial_dim_count](const char dim) { in Convert()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | data_format_ops.cc | 117 const int spatial_dim_count = full_dim_count - 2; in Compile() local
121 input_tensor_shape.num_elements() == spatial_dim_count || in Compile()
124 spatial_dim_count, " or ", in Compile()
129 input_tensor_shape.dim_size(0) == spatial_dim_count || in Compile()
133 spatial_dim_count, " or ", in Compile()
145 if (input_tensor_shape.dim_size(0) == spatial_dim_count) { in Compile()
149 [spatial_dim_count](string* format_str) -> void { in Compile()
152 [spatial_dim_count](const char dim) { in Compile()
154 (spatial_dim_count == 2 || dim != 'D'); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | data_format_ops.cc | 149 const int spatial_dim_count = full_dim_count - 2; in Compute() local
153 input.NumElements() == spatial_dim_count || in Compute()
156 spatial_dim_count, " or ", in Compute()
161 input.dim_size(0) == spatial_dim_count || in Compute()
165 spatial_dim_count, " or ", in Compute()
182 if (input.dim_size(0) == spatial_dim_count) { in Compute()
186 [spatial_dim_count](string* format_str) -> void { in Compute()
189 [spatial_dim_count](const char dim) { in Compute()
191 (spatial_dim_count == 2 || dim != 'D'); in Compute()
197 if (spatial_dim_count == 3) { in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 514 const int64_t spatial_dim_count = spatial_dimensions_to_split.size(); in HaloDuplicateWithSlice() local
516 IPow<int64_t>(ctrl_.number_of_splits, spatial_dim_count); in HaloDuplicateWithSlice()
758 const int64_t spatial_dim_count = spatial_dimensions.size(); in SplitAndTransposeMergedBatch() local
760 for (int64_t i = 0; i < spatial_dim_count; ++i) { in SplitAndTransposeMergedBatch()
769 if (spatial_dim_count > 1) { in SplitAndTransposeMergedBatch()
780 for (int i = 0; i < spatial_dim_count; ++i) { in SplitAndTransposeMergedBatch()
782 batch_dimension + spatial_dim_count - i; in SplitAndTransposeMergedBatch()
784 batch_dimension + spatial_dim_count + 1 + i; in SplitAndTransposeMergedBatch()
803 const int64_t spatial_dim_count = spatial_dimensions.size(); in ChangeSpatialSizeOnSpaceToBatchedShape() local
822 spatial_dim_count); in ChangeSpatialSizeOnSpaceToBatchedShape()
[all …]
|