HomeSort by relevance Sort by last modified time
    Searched full:matrix_shape_01 (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
index_util_test.cc 49 Shape matrix_shape_01 = ShapeUtil::MakeShape(F32, {10, 20}); local
50 SetMinorToMajorLayout(&matrix_shape_01, {0, 1});
53 EXPECT_EQ(0, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01,
55 EXPECT_EQ(199, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01,
57 EXPECT_EQ(53, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01,
60 IndexUtil::LinearIndexToMultidimensionalIndex(matrix_shape_01, 53));

Completed in 202 milliseconds