Searched refs:input_size_t (Results 1 – 1 of 1) sorted by relevance
949 const Tensor* input_size_t = nullptr; in ExtractCudnnRNNParamsInfo() local950 TF_RETURN_IF_ERROR(context->input("input_size", &input_size_t)); in ExtractCudnnRNNParamsInfo()951 if (!TensorShapeUtils::IsScalar(input_size_t->shape())) { in ExtractCudnnRNNParamsInfo()954 int input_size = input_size_t->scalar<int>()(); in ExtractCudnnRNNParamsInfo()1097 const Tensor* input_size_t = nullptr; in Compute() local1098 OP_REQUIRES_OK(context, context->input("input_size", &input_size_t)); in Compute()1099 CHECK(TensorShapeUtils::IsScalar(input_size_t->shape())) in Compute()1101 int input_size = input_size_t->scalar<int>()(); in Compute()