Home
last modified time | relevance | path

Searched refs:expected_dims (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape_test.py320 dict(x_dims=[], y_dims=[], expected_dims=[]),
321 dict(x_dims=[], y_dims=[2], expected_dims=[2]),
322 dict(x_dims=[], y_dims=[2, 3], expected_dims=[2, 3]),
326 expected_dims=[2, (2, 3), (5, 7, 2, 0, 9)]),
328 dict(x_dims=[3], y_dims=[4, 2, 3], expected_dims=[4, 2, 3]),
329 dict(x_dims=[1], y_dims=[4, 2, 3], expected_dims=[4, 2, 3]),
330 dict(x_dims=[3], y_dims=[4, 2, 1], expected_dims=[4, 2, 3]),
334 expected_dims=[3, (2, 3, 1), 3]),
335 dict(x_dims=[1], y_dims=[3, (2, 1, 3)], expected_dims=[3, (2, 1, 3)]),
339 expected_dims=[3, (2, 1, 3), 8]),
[all …]
/external/tensorflow/tensorflow/core/framework/
Dshape_inference_testutil.cc145 auto expected_dims = str_util::Split(expected, ','); in InferShapes() local
147 return Unknown(err_prefix, " expected rank ", expected_dims.size(), in InferShapes()
150 if (c.Rank(out) != expected_dims.size()) { in InferShapes()
151 return Unknown(err_prefix, " expected rank ", expected_dims.size(), in InferShapes()
154 for (int j = 0; j < expected_dims.size(); ++j) { in InferShapes()
156 StringPiece expected_dim(expected_dims[j]); in InferShapes()
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes_test.py146 expected_dims = expected.dims
147 if expected_dims is None:
150 self.assertIsNone(expected_dims)
152 self.assertEqual(len(expected_dims), len(actual_dims))
153 for expected_dim, actual_dim in zip(expected_dims, actual_dims):
/external/tensorflow/tensorflow/core/kernels/
Dtranspose_util_test.cc28 const gtl::ArraySlice<int64>& expected_dims) { in TestDimensionReduction() argument
36 EXPECT_EQ(computed_dims, expected_dims); in TestDimensionReduction()
Dmkl_concat_op.cc186 size_t expected_dims = expected_shape.dims(); in Compute() local
188 if (concat_dim < 0) concat_dim = expected_dims + concat_dim; in Compute()
196 context, s_dims == expected_dims, in Compute()
200 s_dims, " vs. expected rank = ", expected_dims)); in Compute()
202 for (int d = 0; d < expected_dims; ++d) { in Compute()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc195 const std::vector<int>& expected_dims, in ValidateWeights() argument
197 ExpectTrtDimsEqualsArray(expected_dims, weights.shape_); in ValidateWeights()
1336 const std::vector<int>& expected_dims, in TestConvertConst()
1361 ValidateWeights(output.weights(), expected_dims, expected_value); in TestConvertConst()