Home
last modified time | relevance | path

Searched refs:release_output (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc115 output->push_back(op_context.release_output(i)); in EvaluateNode()
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_test.cc86 delete reduction_context->release_output(0).tensor; in BM_SegmentReduction()
Dsparse_to_dense_op_test.cc262 delete sparse_context->release_output(0).tensor; in BM_SparseToDense()
Dnn_ops_test.cc778 delete context->release_output(0).tensor; in BM_LRNFloat()
859 delete avgpool_context->release_output(0).tensor; in BM_AvgPool()
964 delete avgpool_context->release_output(0).tensor; in BM_AvgPoolBk()
1054 delete maxpool_context->release_output(0).tensor; in BM_MaxPool()
1229 delete relu_context->release_output(0).tensor; in BM_ReluFloat()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc181 outputs[o] = op_context.release_output(o); in Compile()
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc295 TensorValue val = ctx.release_output(j); in RunAsync()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h1183 TensorValue release_output(int index);
1577 inline TensorValue OpKernelContext::release_output(int index) { in release_output() function
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc2011 const TensorValue val = ctx->release_output(i); in ProcessOutputs()