Searched defs:key_tensor (Results 1 – 4 of 4) sorted by relevance
521 const Tensor* key_tensor; in Compute() local583 const Tensor* key_tensor; in Compute() local644 const Tensor* key_tensor; in Compute() local
96 Tensor& key_tensor = out_tensors->back(); in GetNextInternal() local
264 Tensor key_tensor(ctx->allocator({}), DT_STRING, {}); in GetNextInternal() local
514 Tensor key_tensor = ctx->input(ctx->num_inputs() - 1); in Compute() local