Searched refs:depth_dim (Results 1 – 6 of 6) 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()
39 EXPECT_EQ(depth_dim, shape_4d->data[3]); 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()
55 EXPECT_EQ(depth_dim, shape_2d->data[1]); 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/eigen/unsupported/test/ |
| D | cxx11_tensor_forced_eval.cpp | 56 Eigen::array<int, 1> depth_dim; in test_const() local
57 depth_dim[0] = 0; in test_const()
65 …Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dim… in test_const()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | Conv3D.cpp | 33 constexpr unsigned int depth_dim = 3u; variable
74 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d()
129 const unsigned int src_depth = src.shape()[depth_dim]; in calculate_conv3d()
199 const unsigned int dst_depth = dst.shape()[depth_dim]; in conv3d()
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 1445 constexpr unsigned int depth_dim = 3u; in compute_conv3d_shape() local
1473 …output_depth_size = static_cast<int>(std::floor((static_cast<float>(src[depth_dim] + pad_front + … in compute_conv3d_shape()
1478 …output_depth_size = static_cast<int>(std::ceil((static_cast<float>(src[depth_dim] + pad_front + p… in compute_conv3d_shape()
1487 output_shape.set(depth_dim, output_depth_size); in compute_conv3d_shape()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | README.md | 319 Tensor<...> Y = ((X - X.maximum(depth_dim).reshape(dims2d).broadcast(bcast))
327 ((X - X.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast))
336 Y = Y / (Y.sum(depth_dim).reshape(dims2d).broadcast(bcast));
342 Y = Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
353 (Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast))).eval();
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1154 DimensionHandle depth_dim = c->Dim( in AvgPoolShape() local
1172 {output_rows, output_cols}, depth_dim, in AvgPoolShape()
|