Searched refs:num_units_t (Results 1 – 1 of 1) sorted by relevance
943 const Tensor* num_units_t = nullptr; in ExtractCudnnRNNParamsInfo() local944 TF_RETURN_IF_ERROR(context->input("num_units", &num_units_t)); in ExtractCudnnRNNParamsInfo()945 if (!TensorShapeUtils::IsScalar(num_units_t->shape())) { in ExtractCudnnRNNParamsInfo()948 int num_units = num_units_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo()1091 const Tensor* num_units_t = nullptr; in Compute() local1092 OP_REQUIRES_OK(context, context->input("num_units", &num_units_t)); in Compute()1093 CHECK(TensorShapeUtils::IsScalar(num_units_t->shape())) in Compute()1095 int num_units = num_units_t->scalar<int>()(); in Compute()