Home
last modified time | relevance | path

Searched refs:set_persistent_memory_size (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Danalytical_cost_estimator.cc66 node->set_persistent_memory_size(node_costs.persistent_memory); in AddCostNode()
H A Dvirtual_scheduler.cc1327 mem_stats->set_persistent_memory_size(persistent_memory_size); in GenerateRunMetadata()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dstep_stats_collector.cc158 ms->set_persistent_memory_size(ctx->persistent_memory_allocated()); in SetMemory()
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dcostmodel.cc539 cnode->set_persistent_memory_size(PersistentMemorySize(n).value()); in AddToCostGraphDef()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dcost_graph.pb.h675 void set_persistent_memory_size(::int64_t value); in Swap()
1478 inline void CostGraphDef_Node::set_persistent_memory_size(::int64_t value) { in set_persistent_memory_size() function
Dstep_stats.pb.h771 void set_persistent_memory_size(::int64_t value); in Swap()
1876 inline void MemoryStats::set_persistent_memory_size(::int64_t value) { in set_persistent_memory_size() function