Searched refs:top_k_ (Results 1 – 2 of 2) sorted by relevance
39 OP_REQUIRES_OK(context, context->GetAttr("top_k", &top_k_)); in BipartiteMatchOp()72 auto valid_top_k = top_k_; in Compute()128 int top_k_; member in tensorflow::BipartiteMatchOp
33 top_k_ = AddInput(TensorType_INT32); in TopKV2OpModel()38 PopulateTensor<int32_t>(top_k_, {top_k}); in TopKV2OpModel()87 int top_k_; member in tflite::__anondc2192160111::TopKV2OpModel