| 1093 | template <typename T> |
| 1094 | T* op_device_context(); |
| 1095 | DeviceContext* op_device_context() { |
| 1096 | DeviceContext* ret = params_->op_device_context; |
| 1097 | if (ret == nullptr) { |
| 1098 | auto* dev_info = device()->tensorflow_gpu_device_info(); |
| 1099 | if (dev_info) ret = dev_info->default_context; |
| 1100 | } |
| 1101 | return ret; |
| 1102 | } |
| 1103 | |
| 1104 | AllocatorAttributes input_alloc_attr(int index) const { |
| 1105 | if (params_->input_alloc_attrs == nullptr) { |
nothing calls this directly
no test coverage detected