Searched defs:keys_tensor (Results 1 – 2 of 2) sorted by relevance
254 const Tensor& keys_tensor = ctx->input(0); in Compute() local
270 const Tensor& keys_tensor = ctx->input(0); in Compute() local