Searched refs:SetReferencedTensors (Results 1 – 4 of 4) sorted by relevance
86 virtual void SetReferencedTensors(const TensorReferenceVector& tensors) = 0;115 void SetReferencedTensors(const TensorReferenceVector& tensors) override;
118 void SetReferencedTensors(NodeExecStatsInterface* stats, in SetReferencedTensors() function121 stats->SetReferencedTensors(tensors); in SetReferencedTensors()1780 nodestats::SetReferencedTensors(stats, accessed); in Process()1862 nodestats::SetReferencedTensors(stats, accessed_tensors); in Process()
160 void NodeExecStatsWrapper::SetReferencedTensors( in SetReferencedTensors() function in tensorflow::NodeExecStatsWrapper
87 void SetReferencedTensors(const TensorReferenceVector& tensors) override {} in SetReferencedTensors() function in tensorflow::data::__anonf8786f2a0111::SimpleStepStatsCollector::SimpleNodeExecStats