Searched refs:MemoryTypesForNode (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | memory_types_test.cc | 57 TEST(MemoryTypesForNode, Simple) { in TEST() argument
72 TF_EXPECT_OK(MemoryTypesForNode(OpRegistry::Global(), DEVICE_CPU, node_def, in TEST()
87 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 | 82 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() function
|
| D | op_kernel.cc | 1654 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()
219 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
|
| D | constant_folding.cc | 523 if (!MemoryTypesForNode(graph->op_registry(), device_type, in ReplaceTensorWithConstant()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | partially_decluster_pass.cc | 73 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
94 TF_RETURN_IF_ERROR(MemoryTypesForNode(graph.op_registry(), device_type, in FindNodesToDecluster()
|
| D | build_xla_ops_pass.cc | 410 TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(), in GetOutputMemoryTypes()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 197 tensorflow::Status s = tensorflow::MemoryTypesForNode( in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 589 TF_RETURN_IF_ERROR(MemoryTypesForNode( in BuildMemoryDeviceInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 389 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|
| D | generic_layout_optimizer_transposer.cc | 147 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|