Searched refs:MemoryTypesForNode (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types_test.cc | 55 TEST(MemoryTypesForNode, Simple) { in TEST() argument
66 TF_EXPECT_OK(MemoryTypesForNode(OpRegistry::Global(), DEVICE_CPU, node_def, in TEST()
80 TF_EXPECT_OK(MemoryTypesForNode(OpRegistry::Global(), DEVICE_GPU, node_def, in TEST()
|
| D | memory_types.h | 31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
|
| D | memory_types.cc | 75 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() function
|
| D | op_kernel.cc | 1318 TF_RETURN_IF_ERROR(MemoryTypesForNode(OpRegistry::Global(), device_type, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types.cc | 66 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
209 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
|
| D | constant_folding.cc | 496 if (!MemoryTypesForNode(graph->op_registry(), device_type, in ReplaceTensorWithConstant()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 61 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
82 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 258 s = tensorflow::MemoryTypesForNode(tensorflow::OpRegistry::Global(), dev_type, node, in TF_GetSupportedDevices()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 562 TF_RETURN_IF_ERROR(MemoryTypesForNode( in BuildMemoryDeviceInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 381 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|