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