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

Method allocate

tensorflow/core/framework/op_kernel.h:1773–1778  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1771}
1772
1773inline Status OpOutputList::allocate(int i, const TensorShape& shape,
1774 Tensor** output) {
1775 DCHECK_GE(i, 0);
1776 DCHECK_LT(i, stop_ - start_);
1777 return ctx_->allocate_output(start_ + i, shape, output);
1778}
1779
1780inline void OpOutputList::set(int i, const Tensor& tensor) {
1781 DCHECK_GE(i, 0);

Callers 15

ComputeMethod · 0.45
InitializeMethod · 0.45
operator()Method · 0.45
TransposeSimpleFunction · 0.45
SpatialConvolutionMethod · 0.45
CuboidConvolutionMethod · 0.45
ComputeMethod · 0.45

Calls 1

allocate_outputMethod · 0.80

Tested by

no test coverage detected