Searched refs:TYPECASE (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 213 #define TYPECASE(dt, X, Y) \ macro
221 TYPECASE(DT_FLOAT, tensor_ptr, dest_ptr); in GetTensorAddress()
222 TYPECASE(DT_HALF, tensor_ptr, dest_ptr); in GetTensorAddress()
223 TYPECASE(DT_INT8, tensor_ptr, dest_ptr); in GetTensorAddress()
224 TYPECASE(DT_INT32, tensor_ptr, dest_ptr); in GetTensorAddress()
|