Searched refs:num_layers_t (Results 1 – 2 of 2) sorted by relevance
937 const Tensor* num_layers_t = nullptr; in ExtractCudnnRNNParamsInfo() local938 TF_RETURN_IF_ERROR(context->input("num_layers", &num_layers_t)); in ExtractCudnnRNNParamsInfo()939 if (!TensorShapeUtils::IsScalar(num_layers_t->shape())) { in ExtractCudnnRNNParamsInfo()942 int num_layers = num_layers_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo()1103 const Tensor* num_layers_t = nullptr; in Compute() local1104 OP_REQUIRES_OK(context, context->input("num_layers", &num_layers_t)); in Compute()1105 CHECK(TensorShapeUtils::IsScalar(num_layers_t->shape())) in Compute()1107 int num_layers = num_layers_t->scalar<int>()(); in Compute()
947 Tensor* num_layers_t = nullptr; in Compute() local949 "num_layers", TensorShape({}), &num_layers_t)); in Compute()963 num_layers_t->scalar<int64>()() = num_layers; in Compute()