OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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