Searched refs:CopyDenseElementsBy (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_utils.cc | 71 void CopyDenseElementsBy(mlir::DenseElementsAttr data, in CopyDenseElementsBy() function
149 CopyDenseElementsBy<bool>(data, output); in CopyDenseElementsDataToXlaFormat()
153 CopyDenseElementsBy<uint8>(data, output); in CopyDenseElementsDataToXlaFormat()
157 CopyDenseElementsBy<uint16>(data, output); in CopyDenseElementsDataToXlaFormat()
161 CopyDenseElementsBy<uint32>(data, output); in CopyDenseElementsDataToXlaFormat()
165 CopyDenseElementsBy<uint64>(data, output); in CopyDenseElementsDataToXlaFormat()
169 CopyDenseElementsBy<bfloat16>(data, output); in CopyDenseElementsDataToXlaFormat()
173 CopyDenseElementsBy<half>(data, output); in CopyDenseElementsDataToXlaFormat()
177 CopyDenseElementsBy<float>(data, output); in CopyDenseElementsDataToXlaFormat()
181 CopyDenseElementsBy<double>(data, output); in CopyDenseElementsDataToXlaFormat()
[all …]
|