Searched refs:num_rows_tensor (Results 1 – 2 of 2) sorted by relevance
205 auto& num_rows_tensor = context->input(2); in Compute() local206 OP_REQUIRES(context, TensorShapeUtils::IsScalar(num_rows_tensor.shape()), in Compute()208 num_rows = num_rows_tensor.flat<int32>()(0); in Compute()
1454 const Tensor* num_rows_tensor = c->input_tensor(2); in MatrixDiagV2Shape() local1458 if (num_rows_tensor != nullptr) { in MatrixDiagV2Shape()1459 TF_RETURN_IF_ERROR(c->GetScalarFromTensor(num_rows_tensor, &num_rows)); in MatrixDiagV2Shape()