Searched refs:width_dim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils_test.cc | 27 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
34 Get4DShape(&batch_dim, &height_dim, &width_dim, &depth_dim, shape_4d), in TEST()
38 EXPECT_EQ(width_dim, shape_4d->data[2]); in TEST()
45 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
50 Get4DShape(&batch_dim, &height_dim, &width_dim, &depth_dim, shape_2d), in TEST()
54 EXPECT_EQ(width_dim, shape_2d->data[0]); in TEST()
61 unsigned int batch_dim, height_dim, width_dim, depth_dim; in TEST() local
64 Get4DShape(&batch_dim, &height_dim, &width_dim, &depth_dim, shape_5d), in TEST()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Conv3D.cpp | 35 constexpr unsigned int width_dim = 1u; variable
72 const unsigned int src_width = src.shape()[width_dim]; in calculate_conv3d()
127 const unsigned int src_width = src.shape()[width_dim]; in calculate_conv3d()
197 const unsigned int dst_width = dst.shape()[width_dim]; in conv3d()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 189 int width_dim = input_shape.dims() - 2; in Compile() local
192 const int64_t width = input_shape.dim_size(width_dim); in Compile()
208 {height_dim, width_dim}); in Compile()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 1447 constexpr unsigned int width_dim = 1u; in compute_conv3d_shape() local
1471 …output_width_size = static_cast<int>(std::floor((static_cast<float>(src[width_dim] + pad_left + p… in compute_conv3d_shape()
1476 …output_width_size = static_cast<int>(std::ceil((static_cast<float>(src[width_dim] + pad_left + pa… in compute_conv3d_shape()
1485 output_shape.set(width_dim, output_width_size); in compute_conv3d_shape()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 1721 const int width_dim = context.src_dim_indices.at('W'); in IsInputConvertible() local
1722 if (shape.dim(height_dim).size() == 1 && shape.dim(width_dim).size() == 1) { in IsInputConvertible()
|