Searched refs:height_seq_tensor (Results 1 – 1 of 1) sorted by relevance
242 const Tensor& height_seq_tensor = context->input(3); in Compute() local296 auto height_seq_tensor_flat = height_seq_tensor.flat<int64_t>(); in Compute()309 for (int64_t hs = 0; hs < height_seq_tensor.dim_size(0) - 1; ++hs) { in Compute()