Searched refs:release_output (Results 1 – 9 of 9) sorted by relevance
94 output->push_back(op_context.release_output(i)); in EvaluateNode()
84 delete reduction_context->release_output(0).tensor; in BM_SegmentReduction()
259 delete sparse_context->release_output(0).tensor; in BM_SparseToDense()
803 delete context->release_output(0).tensor; in BM_LRNFloat()883 delete avgpool_context->release_output(0).tensor; in BM_AvgPool()986 delete avgpool_context->release_output(0).tensor; in BM_AvgPoolBk()1077 delete maxpool_context->release_output(0).tensor; in BM_MaxPool()1251 delete relu_context->release_output(0).tensor; in BM_ReluFloat()1323 delete softplus_context->release_output(0).tensor; in BM_SoftplusFloat()
188 outputs[o] = op_context.release_output(o); in Compile()
424 TensorValue val = ctx.release_output(j); in Run()
1214 TensorValue release_output(int index);1628 inline TensorValue OpKernelContext::release_output(int index) { in release_output() function
734 delete ctx->release_output(i).tensor; in ComputeAsync()
998 const TensorValue val = ctx->release_output(i); in ProcessOutputs()