Searched refs:GetDeviceAndAllocator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 77 TEST(ConvertGraphTest, GetDeviceAndAllocator) { in TEST() argument
82 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
100 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
111 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
128 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
137 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
146 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
|
| D | convert_graph.h | 56 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params,
|
| D | convert_graph.cc | 435 GetDeviceAndAllocator(params, info); in CreateTRTNode()
631 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params, in GetDeviceAndAllocator() function
|