Searched refs:CreateSparse (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 256 TEST_F(LiteralUtilTest, CreateSparse) { in TEST_F() argument
265 auto literal = LiteralUtil::CreateSparse<int64>( in TEST_F()
1959 auto literal = LiteralUtil::CreateSparse<float>({10, 10, 10}, in TEST_F()
1974 LiteralUtil::CreateSparse<bool>(dimensions, indices, {true, false, true}) in TEST_F()
1977 EXPECT_EQ(LiteralUtil::CreateSparse<int64>(dimensions, indices, {1, 2, 3}) in TEST_F()
1981 LiteralUtil::CreateSparse<double>(dimensions, indices, {1.0, 2.0, 3.0}) in TEST_F()
1984 EXPECT_EQ(LiteralUtil::CreateSparse<half>(dimensions, indices, in TEST_F()
1988 EXPECT_EQ(LiteralUtil::CreateSparse<complex64>( in TEST_F()
|
| D | literal_util.h | 140 static Literal CreateSparse(absl::Span<const int64> dimensions,
413 /* static */ Literal LiteralUtil::CreateSparse( in CreateSparse() function
|