Searched refs:GetDeviceAndAllocator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 150 TEST(ConvertGraphTest, GetDeviceAndAllocator) { in TEST() argument
155 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
173 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
184 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
201 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
210 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
219 auto result = GetDeviceAndAllocator(params, engine_info); in TEST()
|
| D | convert_graph.h | 96 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params,
|
| D | convert_graph.cc | 730 std::pair<int, Allocator*> GetDeviceAndAllocator(const ConversionParams& params, in GetDeviceAndAllocator() function
909 auto device_alloc = GetDeviceAndAllocator(params, engine); in ConvertAfterShapes()
|