HomeSort by relevance Sort by last modified time
    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