Searched refs:depth_dim (Results 1 – 4 of 4) 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/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | README.md | 321 Tensor<...> Y = ((X - X.maximum(depth_dim).reshape(dims2d).broadcast(bcast))
329 ((X - X.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast))
338 Y = Y / (Y.sum(depth_dim).reshape(dims2d).broadcast(bcast));
344 Y = Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));
355 (Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast))).eval();
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1099 DimensionHandle depth_dim = c->Dim( in AvgPoolShape() local
1117 {output_rows, output_cols}, depth_dim, in AvgPoolShape()
|