Searched refs:width_dim (Results 1 – 3 of 3) 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/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 182 int width_dim = input_shape.dims() - 2; in Compile() local
185 const int64 width = input_shape.dim_size(width_dim); in Compile()
201 {height_dim, width_dim}); in Compile()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 1726 const int width_dim = context.src_dim_indices.at('W'); in IsInputConvertible() local
1727 if (shape.dim(height_dim).size() == 1 && shape.dim(width_dim).size() == 1) { in IsInputConvertible()
|