Helper function to help setting Callable related parts properly, adapoted from https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/graph_execution_state.cc#L382
| 187 | // adapoted from |
| 188 | // https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/common_runtime/graph_execution_state.cc#L382 |
| 189 | bool |
| 190 | IsGPUFeedAndFetchSupported(TRITONTF_DataType dtype) |
| 191 | { |
| 192 | switch (ConvertDataType(dtype)) { |
| 193 | case tensorflow::DT_BFLOAT16: |
| 194 | case tensorflow::DT_BOOL: |
| 195 | case tensorflow::DT_COMPLEX128: |
| 196 | case tensorflow::DT_COMPLEX64: |
| 197 | case tensorflow::DT_DOUBLE: |
| 198 | case tensorflow::DT_FLOAT: |
| 199 | case tensorflow::DT_HALF: |
| 200 | case tensorflow::DT_INT16: |
| 201 | case tensorflow::DT_INT64: |
| 202 | case tensorflow::DT_INT8: |
| 203 | case tensorflow::DT_UINT16: |
| 204 | case tensorflow::DT_UINT8: |
| 205 | return true; |
| 206 | default: |
| 207 | return false; |
| 208 | } |
| 209 | } |
| 210 | |
| 211 | void |
| 212 | NewSessionOptions( |
no test coverage detected