MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetOutput

Method SetOutput

tensorflow/core/common_runtime/step_stats_collector.cc:163–168  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

161}
162
163void NodeExecStatsWrapper::SetOutput(int slot, const Tensor* tensor) {
164 DCHECK(tensor);
165 NodeOutput* node_output = stats_->add_output();
166 node_output->set_slot(slot);
167 tensor->FillDescription(node_output->mutable_tensor_description());
168}
169
170void NodeExecStatsWrapper::SetReferencedTensors(
171 const TensorReferenceVector& tensors) {

Callers 1

SetOutputFunction · 0.45

Calls 2

add_outputMethod · 0.80
FillDescriptionMethod · 0.45

Tested by

no test coverage detected