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

Method mutable_output

tensorflow/core/framework/op_kernel.h:1689–1695  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1687}
1688
1689inline Tensor* OpKernelContext::mutable_output(int index) {
1690 DCHECK_GE(index, 0);
1691 DCHECK_LT(index, num_outputs());
1692 // No need to record_tensor_reference since the output must already
1693 // have been set by a call that did so.
1694 return outputs_[index].tensor;
1695}
1696
1697inline TensorValue OpKernelContext::release_output(int index) {
1698 DCHECK_GE(index, 0);

Callers 15

TESTFunction · 0.60
SetUpMethod · 0.60
SetOutputExpressionMethod · 0.60
PopulateOutputsMethod · 0.60
ExecuteGraphParallelMethod · 0.60
ExecuteMethod · 0.60
WaitForExecutionMethod · 0.60
TEST_FFunction · 0.60
operator[]Method · 0.60
DoGatherMethod · 0.60
DoReduceMethod · 0.60

Calls

no outgoing calls

Tested by 15

TESTFunction · 0.48
SetUpMethod · 0.48
TEST_FFunction · 0.48
DoGatherMethod · 0.48
DoReduceMethod · 0.48
DoBroadcastMethod · 0.48
DoAddChainMethod · 0.48
GetOutputMethod · 0.48
BM_LRNFloatFunction · 0.48
BM_AvgPoolFunction · 0.48
BM_AvgPoolBkFunction · 0.48