Home
last modified time | relevance | path

Searched refs:SchedulerState (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h321 class SchedulerState {
323 SchedulerState(const bool use_static_shapes,
331 SchedulerState(SchedulerState&& arg) = default;
334 SchedulerState& operator=(SchedulerState&& arg) = delete;
335 SchedulerState(const SchedulerState&) = delete;
336 SchedulerState& operator=(const SchedulerState&) = delete;
340 virtual ~SchedulerState();
464 std::unique_ptr<SchedulerState> scheduler_state);
519 std::unique_ptr<SchedulerState> scheduler_state_;
Dvirtual_scheduler.cc382 SchedulerState::~SchedulerState() {} in ~SchedulerState()
384 SchedulerState::SchedulerState(const bool use_static_shapes, in SchedulerState() function in tensorflow::grappler::SchedulerState
399 Status SchedulerState::Init(const GrapplerItem* item, in Init()
588 void SchedulerState::MaybeUpdateInputOutput(const NodeDef* node) { in MaybeUpdateInputOutput()
637 string SchedulerState::DeviceName(const NodeDef* node) const { in DeviceName()
641 string SchedulerState::SanitizedDeviceName(const NodeDef* node) const { in SanitizedDeviceName()
648 string SchedulerState::ChannelDeviceName(const NodeDef* from, in ChannelDeviceName()
655 std::pair<const NodeDef*, const NodeDef*> SchedulerState::CreateSendRecv( in CreateSendRecv()
745 OpContext SchedulerState::CreateOpContext(const NodeDef* node) const { in CreateOpContext()
777 NodeState& SchedulerState::GetNodeStateOrCreateIt(const NodeDef* node) { in GetNodeStateOrCreateIt()
[all …]