Searched refs:SetTransfer (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler.cc | 1441 void SetTransfer(const string& key, absl::Span<const DataType> types, in SetTransfer() function
1461 SetTransfer(key, types, shapes, &new_transfer); in SetDeviceToHostMetadata()
1470 SetTransfer(key, types, shapes, &transfer); in SetDeviceToHostMetadata()
1495 SetTransfer(key, types, shapes, &new_transfer); in SetHostToDeviceMetadata()
1504 SetTransfer(key, types, shapes, &transfer); in SetHostToDeviceMetadata()
|