Searched refs:max_depth_t (Results 1 – 1 of 1) sorted by relevance
73 const Tensor* max_depth_t; in Compute() local74 OP_REQUIRES_OK(context, context->input("max_depth", &max_depth_t)); in Compute()75 const auto max_depth = max_depth_t->scalar<int32>()(); in Compute()275 const Tensor* max_depth_t; in Compute() local276 OP_REQUIRES_OK(context, context->input("max_depth", &max_depth_t)); in Compute()277 const auto max_depth = max_depth_t->scalar<int32>()(); in Compute()