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