Searched refs:ExtractLiteralFromIndexPositions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1354 Literal ExtractLiteralFromIndexPositions(const Literal& from, in ExtractLiteralFromIndexPositions() function
1378 return ExtractLiteralFromIndexPositions<bool>(from, indices, in ExtractFromIndexPositions()
1382 return ExtractLiteralFromIndexPositions<uint8>(from, indices, in ExtractFromIndexPositions()
1386 return ExtractLiteralFromIndexPositions<int8>(from, indices, in ExtractFromIndexPositions()
1390 return ExtractLiteralFromIndexPositions<bfloat16>(from, indices, in ExtractFromIndexPositions()
1394 return ExtractLiteralFromIndexPositions<Eigen::half>(from, indices, in ExtractFromIndexPositions()
1398 return ExtractLiteralFromIndexPositions<uint16>(from, indices, in ExtractFromIndexPositions()
1402 return ExtractLiteralFromIndexPositions<int16>(from, indices, in ExtractFromIndexPositions()
1406 return ExtractLiteralFromIndexPositions<float>(from, indices, in ExtractFromIndexPositions()
1410 return ExtractLiteralFromIndexPositions<uint32>(from, indices, in ExtractFromIndexPositions()
[all …]
|