Searched refs:ExtractLiteralFromIndexPositions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 2066 Literal ExtractLiteralFromIndexPositions(const Literal& from, in ExtractLiteralFromIndexPositions() function
2090 return ExtractLiteralFromIndexPositions<bool>(from, indices, in ExtractFromIndexPositions()
2094 return ExtractLiteralFromIndexPositions<uint8>(from, indices, in ExtractFromIndexPositions()
2098 return ExtractLiteralFromIndexPositions<int8>(from, indices, in ExtractFromIndexPositions()
2102 return ExtractLiteralFromIndexPositions<bfloat16>(from, indices, in ExtractFromIndexPositions()
2106 return ExtractLiteralFromIndexPositions<Eigen::half>(from, indices, in ExtractFromIndexPositions()
2110 return ExtractLiteralFromIndexPositions<uint16>(from, indices, in ExtractFromIndexPositions()
2114 return ExtractLiteralFromIndexPositions<int16>(from, indices, in ExtractFromIndexPositions()
2118 return ExtractLiteralFromIndexPositions<float>(from, indices, in ExtractFromIndexPositions()
2122 return ExtractLiteralFromIndexPositions<uint32>(from, indices, in ExtractFromIndexPositions()
[all …]
|