HomeSort by relevance Sort by last modified time
    Searched refs:expected_dims (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
transpose_util_test.cc 28 const gtl::ArraySlice<int64>& expected_dims) {
36 EXPECT_EQ(computed_dims, expected_dims);
mkl_concat_op.cc 611 size_t expected_dims = expected_shape.dims(); local
613 if (concat_dim < 0) concat_dim = expected_dims + concat_dim;
626 context, s_dims == expected_dims,
630 s_dims, " vs. expected rank = ", expected_dims));
632 for (int d = 0; d < expected_dims; ++d) {
    [all...]
  /external/tensorflow/tensorflow/core/framework/
shape_inference_testutil.cc 143 auto expected_dims = str_util::Split(expected, ','); local
145 return Unknown(err_prefix, " expected rank ", expected_dims.size(),
148 if (c.Rank(out) != expected_dims.size()) {
149 return Unknown(err_prefix, " expected rank ", expected_dims.size(),
152 for (int j = 0; j < expected_dims.size(); ++j) {
154 StringPiece expected_dim(expected_dims[j]);
  /external/tensorflow/tensorflow/python/framework/
common_shapes_test.py 146 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/contrib/lite/java/src/main/native/
nativeinterpreterwrapper_jni.cc 133 std::unique_ptr<char[]> expected_dims(new char[kBufferSize]);
135 printDims(expected_dims.get(), kBufferSize, target->dims->data,
140 i, expected_dims.get(), obtained_dims.get());

Completed in 227 milliseconds