OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:multi_index
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/
index_util_test.cc
41
std::vector<int64>
multi_index
=
local
43
EXPECT_EQ(1,
multi_index
.size());
44
EXPECT_EQ(42,
multi_index
[0]);
134
std::vector<int64>
multi_index
=
local
137
shape,
multi_index
));
literal_util.cc
198
tensorflow::gtl::ArraySlice<int64>
multi_index
) {
199
return IndexUtil::MultidimensionalIndexToLinearIndex(shape,
multi_index
);
1409
std::vector<int64>
multi_index
;
local
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/tests/
literal_test_util.cc
234
tensorflow::gtl::MutableArraySlice<int64>
multi_index
,
237
NativeT expected_value = expected.Get<NativeT>(
multi_index
);
238
NativeT actual_value = actual.Get<NativeT>(
multi_index
);
246
multi_index
[dimension] = i;
248
expected, actual,
multi_index
, dimension + 1);
280
std::vector<int64>
multi_index
(expected.shape().dimensions_size(), 0);
284
match = ExpectLiteralsEqual<bool>(expected, actual, &
multi_index
, 0);
287
match = ExpectLiteralsEqual<uint8>(expected, actual, &
multi_index
, 0);
290
match = ExpectLiteralsEqual<int32>(expected, actual, &
multi_index
, 0);
293
match = ExpectLiteralsEqual<int64>(expected, actual, &
multi_index
, 0)
515
const auto
multi_index
=
local
[
all
...]
Completed in 92 milliseconds