Home
last modified time | relevance | path

Searched defs:node_state (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler.cc99 const NodeState& node_state, in UpdateDeviceAnnotationState()
602 auto& node_state = node_map_[node]; in MaybeUpdateInputOutput() local
762 const auto& node_state = node_map_.at(node); in CreateOpContext() local
792 auto& node_state = it->second; in GetNodeStateOrCreateIt() local
848 auto& node_state = node_map_[node]; in GetOutputNodes() local
876 auto& node_state = node_map_[node]; in MarkNodeExecuted() local
1354 const auto& node_state = node_map_.at(node); in GetPersistentMemoryUsage() local
1363 auto& node_state = node_map_.at(node); in SetNodeStateTimeScheduled() local
1368 int64_t SchedulerState::GetOrCalculateOutputSize(const NodeState& node_state, in GetOrCalculateOutputSize()
H A Dvirtual_scheduler_test.cc2713 const NodeState& node_state = nodedef_node_state.second; in TEST_F() local
/aosp_15_r20/external/coreboot/src/vendorcode/cavium/bdk/libbdk-hal/
H A Dbdk-nic.c420 nic_node_state_t *node_state = global_node_state[node]; in if_process_complete_rx() local
693 nic_node_state_t *node_state = global_node_state[handle->node]; in bdk_nic_port_init() local
928 const nic_node_state_t *node_state = global_node_state[handle->node]; in bdk_nic_transmit() local
1036 const nic_node_state_t *node_state = global_node_state[handle->node]; in bdk_nic_get_queue_depth() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dtraversal.cc61 absl::flat_hash_map<int, NodeState> node_state; in DfsTraversal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpending_counts.h121 NodeState node_state(Handle h) { in node_state() function
H A Dpropagator_state.h165 PendingCounts::NodeState node_state(PendingCounts::Handle h) { in node_state() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_mixed_precision.cc830 absl::flat_hash_map<int, NodeState> node_state; in DfsTypeTraversal() local