Searched refs:MemoryTypesForNode (Results 1 – 12 of 12) 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 | 81 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() function
|
| D | op_kernel.cc | 1603 TF_RETURN_IF_ERROR(MemoryTypesForNode(OpRegistry::Global(), device_type, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | memory_types.cc | 67 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
216 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 | 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 | 406 TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(), in GetOutputMemoryTypes()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 196 tensorflow::Status s = tensorflow::MemoryTypesForNode( in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 587 TF_RETURN_IF_ERROR(MemoryTypesForNode( in BuildMemoryDeviceInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 388 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|
| D | generic_layout_optimizer_transposer.cc | 133 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|