Searched refs:MemoryTypeForOutput (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types_test.cc | 71 TEST(MemoryTypeChecker, MemoryTypeForOutput) { in TEST() argument
80 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_CPU, g, sf, 0, &memory_type)); in TEST()
85 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_GPU, g, si, 0, &memory_type)); in TEST()
91 TF_EXPECT_OK(MemoryTypeForOutput(DEVICE_SYCL, g, si, 0, &memory_type)); in TEST()
|
| D | memory_types.h | 43 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g,
|
| D | memory_types.cc | 205 Status MemoryTypeForOutput(const DeviceType& device_type, const Graph* g, in MemoryTypeForOutput() function
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 145 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node, in InsertNodes()
|