Searched refs:max_depth_t (Results 1 – 1 of 1) sorted by relevance
84 const Tensor* max_depth_t; in Compute() local85 OP_REQUIRES_OK(context, context->input("max_depth", &max_depth_t)); in Compute()86 const auto max_depth = max_depth_t->scalar<int32>()(); in Compute()