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

  /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/core/kernels/
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...]

Completed in 66 milliseconds