Searched refs:MemoryTypesForNode (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A 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()
|
| H A D | memory_types.h | 31 Status MemoryTypesForNode(const OpRegistryInterface* op_registry,
|
| H A D | memory_types.cc | 82 Status MemoryTypesForNode(const OpRegistryInterface* op_registry, in MemoryTypesForNode() function
|
| H A D | op_kernel.cc | 1722 TF_RETURN_IF_ERROR(MemoryTypesForNode(OpRegistry::Global(), device_type, in CreateOpKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | memory_types.cc | 68 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, in ProcessMemoryTypes()
221 TF_RETURN_IF_ERROR(MemoryTypesForNode(g->op_registry(), device_type, n->def(), in MemoryTypeForOutput()
|
| H A D | constant_folding.cc | 520 if (!MemoryTypesForNode(graph->op_registry(), device_type, in ReplaceTensorWithConstant()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A 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()
|
| H A D | build_xla_ops_pass.cc | 409 TF_RETURN_IF_ERROR(MemoryTypesForNode(root.graph()->op_registry(), in GetOutputMemoryTypes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | cluster_wrapper.cc | 197 tensorflow::Status s = tensorflow::MemoryTypesForNode( in PYBIND11_MODULE()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_partition.cc | 588 TF_RETURN_IF_ERROR(MemoryTypesForNode( in BuildMemoryDeviceInfo()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 147 s = tensorflow::MemoryTypesForNode(OpRegistry::Global(), device_type, in IsHostMemory()
|