Searched refs:max_num_elements_t (Results 1 – 1 of 1) sorted by relevance
226 const Tensor& max_num_elements_t = ctx->input(1); in Compute() local228 ctx, TensorShapeUtils::IsScalar(max_num_elements_t.shape()), in Compute()231 "but got shape: ", max_num_elements_t.shape().DebugString())); in Compute()238 empty.max_num_elements = max_num_elements_t.scalar<int32>()(); in Compute()