Searched refs:output_tensor_array (Results 1 – 1 of 1) sorted by relevance
120 TensorArray* output_tensor_array; in Compute() local122 &output_tensor_array)); in Compute()124 ctx->set_output_ref(0, output_tensor_array->mu(), in Compute()125 output_tensor_array->handle()); in Compute()127 ctx->set_output(0, *output_tensor_array->handle()); in Compute()132 output_tensor_array->resource_handle(ctx); in Compute()150 TensorArray** output_tensor_array) = 0;183 TensorArray** output_tensor_array) override { in CreateTensorArray() argument215 *output_tensor_array = tensor_array; in CreateTensorArray()282 TensorArray** output_tensor_array) override { in CreateTensorArray() argument[all …]